选择 2D PyTorch 张量的特定行

【问题标题】：Select specific rows of 2D PyTorch tensor选择 2D PyTorch 张量的特定行
【发布时间】：2020-05-18 18:53:00
【问题描述】：

假设我有一个看起来像这样的二维张量：

[[44, 50, 1, 32],
.
.
.
[7, 13, 90, 83]]

以及我要选择的行索引列表，类似于 [0, 34, 100, ..., 745]。如何创建一个只包含索引包含在数组中的行的新张量？

【问题讨论】：

标签： python indexing pytorch tensor

【解决方案1】：

你可以用 numpy 选择喜欢

import torch
x = torch.Tensor([[1, 2, 3, 4],
                  [5, 6, 7, 8],
                  [9, 8, 7, 6],
                  [5, 4, 2, 1]])

indices = [0, 3]

print(x[indices])
# tensor([[1., 2., 3., 4.],
#         [5., 4., 2., 1.]])

【讨论】：