#python #pytorch #tensor
Вопрос:
У меня есть модель PyTorch, и я делаю на ней прогноз. После выполнения прогноза я получаю результат в виде
tensor([[-3.4333]], grad_fn=<AddmmBackward>)
Но мне это нужно как обычное целое -3.4333 число . Как я могу это сделать.
Ответ №1:
Вызовите .item свой тензор, чтобы преобразовать его в стандартный номер python.
Ответ №2:
Чтобы получить значение внутри вызова тензора: tensor.item()
Экс:
>>> x = torch.tensor([1.0])
>>> x.item()
1.0
https://pytorch.org/docs/stable/generated/torch.Tensor.item.html