其作用相当于增加一个维度
import torch as t
a = t.randn((128,))
a.shape
>>>torch.Size([128])
a[:, None].shape
>>>torch.Size([128, 1])
其作用相当于增加一个维度
import torch as t
a = t.randn((128,))
a.shape
>>>torch.Size([128])
a[:, None].shape
>>>torch.Size([128, 1])