[Pytorch] pytorch에서 value count 하는 법

Pandas 내에는 value_count() 라는 함수를 통해 데이터 프레임 내의 값을 카운팅 할 수 있다. Numpy 에서는 collection.Counter 등의 함수를 이용하여 몇 개가 있는지를 확인할 수 있다. Pytorch에서는 bincount라는 함수로 대체할 수 있다.

 

import torch 

samples = torch.tensor([1, 1, 2, 1, 3, 3, 3], dtype=torch.long)
torch.bincount(samples)

# tensor([0, 3, 1, 3])

samples.dtype
# torch.int64

당연하게도, LongTensor 인 경우에만 가능하다. FloatTensor의 경우 갯수를 확인하는 것이 불가능하기 때문에, LongTensor로 제대로 데이터 타입이 되어 있는지 확인이 필요하다.