数组切片中的None
程序员文章站
2022-03-03 20:12:13
...
其作用相当于增加一个维度
import torch as t
a = t.randn((128,))
a.shape
>>>torch.Size([128])
a[:, None].shape
>>>torch.Size([128, 1])
您现在的位置是：
首页