pytorch——张量索引和切片

张量中的元素可以通过索引访问：第一个元素的索引是0，最后一个元素的索引是-1，可以指定范围以包含第一个元素和最后一个元素之间的元素。

x = torch.arange(12, dtype=torch.float32).reshape(3, 4)
print(f'x: {x}')
print(f'x[-1]: {x[-1]}')
print(f'x[1:3]: {x[1:3]}')

有三个轴的张量的索引（根据第一个轴进行索引）：

y=torch.arange(36).reshape(3,3,4)
print(f'y: {y}')
print(f'y[-1]: {y[-1]}')
print(f'y[1:3]: {y[1:3]}')

索引指定位置的多个张量，其中“：”代表沿某个轴的所有元素：

print(f'x[0:2,:]: {x[0:2,:]}')
print(f'y[0:2,:,0:2]: {y[0:2,:,0:2]}')