pytorch怎么统计元素出现次数

在PyTorch中，可以使用torch.unique函数来统计元素出现的次数。以下是一个示例代码：

import torch  # 创建一个包含重复元素的张量 x = torch.tensor([1, 2, 2, 3, 3, 3, 4, 4, 4, 4])  # 使用torch.unique函数来获取唯一元素和它们出现的次数 unique_elements, counts = torch.unique(x, return_counts=True)  # 打印结果 for i in range(len(unique_elements)):     print(f"元素 {unique_elements[i]} 出现了 {counts[i]} 次") 

运行以上代码，将得到输出：

元素 1 出现了 1 次 元素 2 出现了 2 次 元素 3 出现了 3 次 元素 4 出现了 4 次