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

PyTorch method for returning element with highest count in a 1D tensor?

是否有内置的 PyTorch 方法接受一维张量,return是张量中计数最高的元素?

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

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

torch.mode() 是内置函数(读取 here),它可以处理您的两种情况。

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