关于numpy中切片的一些讨论

关于用冒号切片

对于这个向量空间:

n = torch.randn(1000, 100, 1, 1)
_____________________________________
test = n[0,:,:,:]

test.shape
Out[10]: torch.Size([100, 1, 1])

test = n[:1,:,:,:]

test.shape
Out[12]: torch.Size([1, 100, 1, 1])

所以说用冒号的切片是可以保留一维空间的