我正在使用 Flask 进行推理,得到这个结果。他们有什么方法可以将此张量转换为浮点数,因为我想使用此结果在反应应用程序中显示
{
result: {
predictions: "tensor([[-3.4333]], grad_fn=<AddmmBackward>)"
}
}
Run Code Online (Sandbox Code Playgroud)
来自Torch.Tensor.item文档:
x = torch.tensor([1.0])
print((x.item())
Run Code Online (Sandbox Code Playgroud)
输出:
1.0
Run Code Online (Sandbox Code Playgroud)
类型检查:
print(type(x.item())
Run Code Online (Sandbox Code Playgroud)
输出:
float
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
12627 次 |
| 最近记录: |