溫馨提示×

pytorch怎么統計元素出現次數

小億
224
2024-03-28 14:37:11
欄目: 深度學習

在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 次

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女