torch配列にある数が何個あるか検知したいです
0が20個あれば20
1が100個あれば100
collections.Counterのようなものです
tensor型の場合collections.Counterがうまく機能しないため
tensor型のままある数が何個あるかを検知する関数はありませんか?(かなりでかい配列なためforで数えたりnumpyに返還したり以外で)
わかる人教えてください
import torch a=torch.tensor([0,0,0,1,1,2,2,2,2,3,3,3,3]) import collections c = collections.Counter(a) print(c) 結果 Counter({tensor(0): 1, tensor(0): 1, tensor(0): 1, tensor(1): 1, tensor(1): 1, tensor(2): 1, tensor(2): 1, tensor(2): 1, tensor(2): 1, tensor(3): 1, tensor(3): 1, tensor(3): 1, tensor(3): 1})
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。