小编pat*_*ick的帖子

Pytorch中的列选择是否可以区分？例如，如果我想从每一行中选择一列以创建一个新的行X 1数组，然后使用此新数组作为背景，背景将正常工作吗？

qvalues = qvalues[range(5),[0,1,0,1,0]]

如上所示从5 * 2张量完成元素选择？