有没有办法在多维张量上获得 torch.mode

Question

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

例如

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

假设你有

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

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

不使用循环

Answer 1

使用torch中的dimension属性来选择应该使用模式操作符减少哪个维度。

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

会给你想要的答案。