torch,选择某几列组成新矩阵

可以使用PyTorch中的索引操作来选择矩阵的某几列，然后将它们组成一个新的矩阵。具体实现如下：

假设原矩阵为`x`，需要选择的列号为`indices`，则可以使用以下代码来得到新矩阵`y`：

import torch

x = torch.randn(3, 5)  # 原矩阵
indices = [0, 2, 4]   # 需要选择的列号

y = x[:, indices]     # 选择某几列组成新矩阵

print(y)              # 输出新矩阵

上述代码中，`x[:, indices]`表示选择`x`矩阵的所有行，以及列号为`indices`的列，最终组成新的矩阵`y`。