pytorch已知张量和某一维度的索引,按索引取出张量中的值

可以使用 PyTorch 的索引功能进行操作，例如：

import torch

# 已知张量
tensor = torch.tensor([[1,2,3],[4,5,6],[7,8,9]])

# 某一维度的索引
dim_index = 1

# 按索引取出张量中的值
result = tensor[:, dim_index]

其中 `:` 表示选取所有行，`dim_index` 表示选取对应维度的索引位置上的值。