Tensor.item()

torch.Tensor.item()

Tensor.item() -> number

• tenosr의 value를 반환한다.
• 1개의 원소로 이루어진 tensor에만 사용할 수 있다.

예시: loss 값 반환

import torch
import torch.nn as nn


y = torch.tensor([0.3, 0.5, 0.9])
pred = torch.tensor([0.0, 0.0, 1.0])
loss = nn.BCELoss(y, pred)
loss # torch.tenosr(0.3851)
loss.item() # 0.3850608766078949