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

Question

I am using flask to do inference and I am getting this result. Is their any way to convert this tensor into float because I want to use this result to display in a react app

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

Answer 1

From Torch.Tensor.item docs:

x = torch.tensor([1.0])

print((x.item())

output:

1.0

type check:

print(type(x.item())

output:

float

Answer 2

You can also use tensor.to(torch.float)