torch.Tensor.item

https://pytorch.org/docs/stable/generated/torch.Tensor.item.html

torch.Tensor.item — PyTorch 2.1 documentation

 

모델에서 예측한 값(y_pred)과 참 값(y_trgt)값이 같은 것의 개수를 구하려 할 때 (y_pred==y_trgt).sum().item()을 쓰는 코드를 발견해서 디버깅을 해보면서 torch.Tensor.item()이 텐서 값을 숫자만으로 나타낼 수 있다는 것을 알았다.

 

이는 dictionary에서 (key, value) 값으로 나타내는 items과 다른 의미를 나타낼 수 있다는 것을 알았다.

 

python items code

car = {
  "brand": "BMW",
  "model": "BMWi7",
  "year": 2024
}

x = car.items()

print(x)

# dict_items([('brand', 'BMW'), ('model', 'BMW i7'), ('year', 2024)])

 

pytorch code

device = 'cpu'
y_pred = tensor([2,8,3,1,4,5])
y_trgt = tensor([3,8,3,1,5,5])

ex = += (y_pred==y_trgt).sum() # tensor(4, device='cpu')
correct += (y_pred==y_trgt).sum().item() # 4