None作为ndarray或tensor的索引

None作为ndarray或tensor的索引作用是增加维度,与 pytorch中的 torch.unsqueeze() 或 tensorflow 中的tf.expand_dims() 作用相同

例子:

In [5]: t=torch.from_numpy(np.arange(12).reshape(3,4))

In [6]: t
Out[6]:
tensor([[ 0,  1,  2,  3],
        [ 4,  5,  6,  7],
        [ 8,  9, 10, 11]])

In [7]: t.dim()
Out[7]: 2

In [8]: t[:,None,:]
Out[8]:
tensor([[[ 0,  1,  2,  3]],

        [[ 4,  5,  6,  7]],

        [[ 8,  9, 10, 11]]])

In [9]: arr=np.arange(12).reshape(3,4)

In [10]: arr
Out[10]:
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])

In [11]: arr[:,None,:]
Out[11]:
array([[[ 0,  1,  2,  3]],

       [[ 4,  5,  6,  7]],

       [[ 8,  9, 10, 11]]])

In [12]: arr[:,None,None,:]
Out[12]:
array([[[[ 0,  1,  2,  3]]],


       [[[ 4,  5,  6,  7]]],


       [[[ 8,  9, 10, 11]]]])

In [13]: arr[:,None,None,:].shape
Out[13]: (3, 1, 1, 4)

In [14]: t
Out[14]:
tensor([[ 0,  1,  2,  3],
        [ 4,  5,  6,  7],
        [ 8,  9, 10, 11]])

In [15]: t.unsqueeze(1)
Out[15]:
tensor([[[ 0,  1,  2,  3]],

        [[ 4,  5,  6,  7]],

        [[ 8,  9, 10, 11]]])

In [16]: t
Out[16]:
tensor([[ 0,  1,  2,  3],
        [ 4,  5,  6,  7],
        [ 8,  9, 10, 11]])

In [17]: t.unsqueeze(1)==t[:,None,:]
Out[17]:
tensor([[[True, True, True, True]],

        [[True, True, True, True]],

        [[True, True, True, True]]])


  • 5
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值