富校长:
总结:(如果":"的数字和维度不够,就会自动往后补,所以相当于后面的:可以省略:
a=torch.Tensor([1,2,3,4,5,6]).reshape(2,3)
print(a.shape)
print(a[:,None].shape)
print(a[:,None,:].shape)
print(a[None,:].shape)
print(a[None,:,:].shape)
print("\r\n")
print(a[None,None,:,:].shape)
print(a[None,None].shape)
print(a[:,:,None].shape)
print(a[:,:,None,None].shape)