如何选取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

# 选取第二组、第