PyTorch torch.mode()：求众数（附带实例）

使用 torch.mode() 函数求众数，该函数返回众数，有以下两种格式：

torch.mode(input, dtype=None)
torch.mode(input, dim, keepdim=False, dtype=None)

请看以下示例。

1) 设置参数 input 和 dim，代码如下：

# 导入torch库
import torch
# 创建一个张量a
a = torch.tensor([[1, 2], [3, 4]])
# 计算张量a所有元素的众数，并将结果打印出来
print(torch.mode(a))
# 沿着第0维计算张量a的元素众数，并将结果打印出来
print(torch.mode(a, 0))

输出结果如下：

torch.return_types.mode(
values=tensor([1, 3]),
indices=tensor([0, 0]))
torch.return_types.mode(
values=tensor([1, 2]),
indices=tensor([0, 0]))

2) 设置参数 keepdim，代码如下：

# 导入torch库
import torch
# 创建一个张量a
a = torch.tensor([[1, 2], [3, 4]])
# 沿着第1维计算张量a的元素众数，并保持原始维度，将结果赋值给变量mode_result
mode_result = torch.mode(a, 1, keepdim=True)
# 打印mode_result的值
print(mode_result)

输出结果如下：

torch.return_types.mode(
values=tensor([[1],[3]]),
indices=tensor([[0],[0]]))