张量中的元素可以通过索引访问:第一个元素的索引是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]}')