如何选取Pytorch的三维tensor中的某个数据

对于一个形状为 (a, b, c) 的三维 tensor，可以使用如下代码选取其中的某个数据：

import torch

x = torch.randn(2, 3, 4)

# 选取第一组、第二行、第三列的数据
print(x[0, 1, 2])  # 输出一个标量

# 选取第一组的数据
print(x[0])  # 输出一个形状为 (3, 4) 的二维 tensor

# 选取第一组、第二组的数据
print(x[:2])  # 输出一个形状为 (2, 3, 4) 的三维 tensor

# 选取第二组、第一行、所有列的数据
print(x[1, 0, :])  # 输出一个形状为 (4,) 的一维 tensor

其中方括号中的数字表示要选取的维度的下标，可以使用冒号来表示选取该维度的所有数据。注意，如果选取的维度不是最后一个维度，则需要在选取完该维度之后继续指定下一个要选取的维度。