PyTorch 方法返回一维张量中计数最高的元素？

Question

是否有内置的 PyTorch 方法接受一维张量，并返回张量中计数最高的元素？

例如，如果我们输入torch.tensor([2,2,2,3,4,5])，该方法应该返回2出现次数最多的值。如果频率相同，则应返回具有较低值的元素；输入torch.tensor([1,1,2,2,4,5])应该返回1.

需要明确的是，我只想知道是否有一个现有的内置 PyTorch 方法可以完全执行此操作。如果没有这样的方法，请不要发布解决方案，因为我想尝试自己解决它。

Answer 1

yestorch.mode()是内置函数（请阅读此处），它可以处理您的两个条件。

torch.mode(alpha,0) #alpha being the name of tensor