pytorch 众数

PyTorch没有直接提供计算众数的函数。要计算众数，可以先使用torch.flatten()将张量展平为一维数组，然后使用torch.mode()函数找到最常出现的元素。下面是一个例子代码可以计算PyTorch张量的众数：

import torch

# 定义一个张量
a = torch.Tensor([1, 2, 3, 4, 4, 5, 5, 5])

# 将张量展平为一维数组
a_flat = torch.flatten(a)

# 使用torch.mode()函数找到最常出现的元素
mode_value = torch.mode(a_flat).values.item()

print("众数为:", mode_value)
输出:
众数为: 5

这段代码首先将张量展平为一维数组，然后使用torch.mode()函数找到最常出现的元素，最后打印出众数。在这个例子中，众数为5。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* *2* *3* [[PyTroch系列-9]：PyTorch基础 - 张量元素的统计运算](https://blog.csdn.net/HiWangWenBing/article/details/119443802)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 100%"]
[ .reference_list ]