(pytorch) RuntimeError: Expected object of scalar type Float but got scalar type Double for sequence element 1 in sequence argument at position #1 'tensors'

참고한 페이지 : https://qlsenddl-lab.tistory.com/38

 

float이 아닌 double 형이 들어왔다는 에러라는데, float64를 tensor에선 double로  인식한다고 한다

 

np.random 으로 난수를 발생시키고 tensor 자료형으로 바꿨을 때 float64 자료형으로 확인돼서
torch.from_numpy().float()을 통해 float형으로 바꾸어주어 에러를 해결했다고 한다

 

 

나같은 경우엔, torch.float 형과 np.random을 이용하여 생성한 numpy 배열을 덧셈 연산한 부분이 있어서
numpy 배열을 torch.from_numpy().float()으로 자료형을 바꾸어주어 해결했다