如何將火炬張量轉換為浮動？

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)