如何将火炬张量转换为浮动？

Question

我正在使用 flask 进行推理，我得到了这个结果。 他们有什么方法可以将这个张量转换为浮点数，因为我想用这个结果在反应应用程序中显示

{
result: {
predictions: "tensor([[-3.4333]], grad_fn=<AddmmBackward>)"
}
}

Answer 1

来自Torch.Tensor.item文档：

x = torch.tensor([1.0])

print((x.item())

输出：

1.0

类型检查：

print(type(x.item())

输出：

float

Answer 2

你也可以使用 tensor.to(torch.float)