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