我正在使用 flask 做推理，我得到这个结果。他们有没有办法把这个Tensor转换成浮点数，因为我想用这个结果显示在一个React应用程序
{ result: { predictions: "tensor([[-3.4333]], grad_fn=<AddmmBackward>)" } }
pobjuy321#
来自Torch. Tensor.项目文件：
x = torch.tensor([1.0]) print((x.item())
输出：
1.0
类型检查：
print(type(x.item())
float
hwamh0ep2#
您也可以使用tensor.to（torch.float）
2条答案
按热度按时间pobjuy321#
来自Torch. Tensor.项目文件：
输出：
类型检查：
输出：
hwamh0ep2#
您也可以使用tensor.to（torch.float）