【Pytorch】None用于扩展维度

使用冒号时,None在第几位,就在那一维上扩展一个维度;如果要扩展多个维度,把冒号视作原始维度,None插入所需的位置即可,最后一个None后面的冒号可以省略不写。

使用省略号,在最后一维扩展维度。

t = torch.tensor([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]])  # (2,2,3)

t1 = t[None]                  # (1,2,2,3)
t2 = t[:, None]               # (2,1,2,3)
t3 = t[:, :, None]            # (2,2,1,3)
t4 = t[:, :, :, None]         # (2,2,3,1)

tt1 = t[:, :, None, :, None]  # (2,2,1,3,1)
tt2 = t[:, :, None, None]     # (2,2,1,1,3)

t5 = t[..., None]             # (2,2,3,1)
tt3 = t[..., None, None]      # (2,2,3,1,1)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值