mat1 and mat2 must have the same dtype, but got Double and Float 에러

 

float 타입이 들어와야하는데 double type이 들어오는 경우 생기는 에러이다. 

dtype 을 본결과 torch.float64 라고 보인다. 이것을 .float() 을 해줘야된다. 

 

이렇게 바꿨더니 에러가 없어짐