x.item() 스칼라 변환 오류

[shawrby]

y = x.item() # x는 텐서, y는 스칼라입니다.

로 했는데.. 로컬에선 문제 없는데, nsml 사용시 발생하네요. 혹시 해결 방안이 있나요?

[nsml-user]

원래 2차원 이상의 텐서는 .item()을 못쓰는 걸로 되어있습니다. nsml에서만 그런건 아닌거 같구요. 2차원 이상의 텐서에 대해서는 .tolist()를 쓰면 list로 반환되니 시도해보시길 바랍니다.