RuntimeError: expected scalar type long but found float
넘파이는 연산하려는 두 행렬의 타입이 달라도 알아서 캐스팅을 해줘서 오류가 나지 않는다. 그러나 pytorch에서는 두 행렬의 타입이 틀리면 오류가 뜨게 된다.
매트릭스의 원소들 타입 검사하려면: a1.dtype 메서드를 사용해 타입을 검사한다.
a1.type(torch.float32)
위의 명령어를 통해서 a1 행렬의 타입을 변경해줄 수 있다.
Study archive