[pytorch] item() 함수

tensor에 하나의 값만 존재한다면, item() 함수를 사용해 python scalar 값을 얻을 수 있다.

tensor에 여러개의 값이 존재한다면, 사용이 불가능하다.

 

아래 예시 코드를 참고해보자.

import torch
x = torch.randn(1)
print(x)
print(x.item())

결과는 다음과 같은 것을 알 수가 있다.