在PyTorch中,可以使用torch.unique
函數(shù)來(lái)統(tǒng)計(jì)元素出現(xiàn)的次數(shù)。以下是一個(gè)示例代碼:
import torch
# 創(chuàng)建一個(gè)包含重復(fù)元素的張量
x = torch.tensor([1, 2, 2, 3, 3, 3, 4, 4, 4, 4])
# 使用torch.unique函數(shù)來(lái)獲取唯一元素和它們出現(xiàn)的次數(shù)
unique_elements, counts = torch.unique(x, return_counts=True)
# 打印結(jié)果
for i in range(len(unique_elements)):
print(f"元素 {unique_elements[i]} 出現(xiàn)了 {counts[i]} 次")
運(yùn)行以上代碼,將得到輸出:
元素 1 出現(xiàn)了 1 次
元素 2 出現(xiàn)了 2 次
元素 3 出現(xiàn)了 3 次
元素 4 出現(xiàn)了 4 次