python - 如何交换 PyTorch Tensor 中的值？

Question

这个问题已经有答案了:

How two rows can be swapped in a torch tensor? (4 个回答)

已关闭 2 年前。

我有以下张量:

vector = torch.tensor([[1,5,3], [2,3,4]])

如何交换第二个轴中的值？例如

tensor([[1, 5, 3], [2, 3, 4]])

变成:

tensor([[1, 3, 3], [2, 5, 4]])

Answer 1

您可以使用 numpy 的索引样式:

>>> vector = torch.tensor([[1,5,3], [2,3,4]])
tensor([[1, 3, 4],
        [2, 5, 3]])
>>> vector[[0,1],1] = vector[[1,0],1]
>>> vector 
tensor([[1, 3, 3],
        [2, 5, 4]])

在这种情况下，我们会在第一个维度的第 0 个索引和第一个索引之间切换值，仅在第二个维度的索引 1 处切换值。