pytorch张量的索引和切片操作

直接上代码

import torch

a = torch.randn(2, 3, 4)  # 构建维度为2×3x4的张量
print(a)
print(a[1, 2, 3])  # 取张量在0维的第一1号、1维的第2号、2维3号的元素（注意编号是从0开始）

print(a[:, 1:-1, 1:3])  # 仅仅一个冒号表示取所有的，-1表示最后一个元素

a[1, 2, 3] = -5  # 直接更改索引和切片会更改原始量的值
print(a)

print(a > 0)  # 大于0为True,小于0为False。输出一个由True,False组成的2*3*4的矩阵

print(a[a>0]) # 输出a中大于0的元素，生成为一维向量

Done!!!