[Pytorch] Tensor 에서 특정 index 의 값을 뽑아 새로운 tensor 정의

torch.where

 

torch.index_select

#class_value 라는 label을 가지는 값들의 index 뽑기

target_indexes = torch.where(target_tensor == class_value)[0]

#뽑은 index의 값을 가지는 새로운 tensor 정의

input_with_class_value = torch.index_select(input_tensor, 0, target_indexes)