torch.scatter 보다 편하게 one hot encoding 값을 설정하는 방법

torch.scatter 보다 편하게 one hot encoding 값을 설정하는 방법

조금 더 직관적인 방법에 대해 설명하고자 합니다.

torch 뿐 아니라 numpy에서도 가능하다는 것을 확인하였고

위 방법을 사용하면 scatter 보다 직관적으로 사용하는 것을 확인할 수 있습니다.

방법은 우리가 원하는 t 값을 설정을 합니다.

그 후 zeros tensor에 t의 길이만큼을 range(len(t))을 행으로 넣고 그 후 t를 열의 값으로 넣은 후 그 값을 1로 하겠다라고 하면 위와 같은 방식을 할 수 있다.

코딩을 잘 하시는 분은 이미 눈치챘겠지만 이것은 for문을 돌린 것입니다.

1줄로 바꾼 것을 위 방식인 것이지요. 

그러나 저 같은 초보자에게는 직관적으로 이해할 수 있음을 볼 수 있습니다.

이후 scatter_ 방식을 쓸지 이 방식을 쓸지는 여러분의 선택~