AI 공부 도전기

IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number


Pytorch에서 코드를 돌리다보면 위와 같은 tensor error를 발견하게 된다.

그럼 조용히 위 에러 아래를 확인하여 에러가 난 위치로 이동

tensor.data[0] 혹은 tensor.data의 형태 혹은 tensor.~의 형태를

tensor.item()으로 바꿔주면 해결된다.


버전이 바뀌면서 에러가 생겼을 가능성이 있음

공유하기

facebook twitter kakaoTalk kakaostory naver band
loading