python中的张量类似于多维数组,所以切片取值的时候要注意他的维度
例如
x = torch.rand(5,5)
print(x)
print(x[:,3])
上面的x为一个二维张量,每一个维度的大小为5
执行结果如下
可以看到取到了x的第三列的所有值
再来一个例子
y = torch.rand(5,5,5)
print(y)
print(y[1:3,:,:])
执行结果
取到的切片值
所以对于一个张量m,m[:,:,:....],m是多少维度的就有多少个被逗号隔开的值
从第一维开始顺序往后,需要取哪一维的那一部分值,就可以在:的两边加上切片范围