有什么办法可以在多维张量上获取torch.mode

Question

有什么方法可以将torch.mode应用于多个维度

例如

import numpy as np
import torch
x = np.random.randint(10, size=(3, 5))
y = torch.tensor(x)

让我们说y有

[[6 3 7 3 0]
 [2 5 7 9 7]
 [6 1 4 6 3]]

torch.mode应该返回大小为3的张量[3,7,6]

不使用循环

Answer 1

使用割炬中的Dimensions属性选择应使用模式运算符减小的尺寸。

torch.mode(y, dim = 1)[0]

会给你想要的答案。