C++ torch::Tensor 转为float
时间: 2023-03-25 18:01:29 浏览: 391
详解torch.Tensor的4种乘法
5星 · 资源好评率100%
可以使用tensor.item()方法将一个只有一个元素的tensor转为float类型,如果有多个元素,则需要使用tensor.tolist()方法将tensor转为list,再使用list comprehension将每个元素转为float类型。例如:
```python
import torch
tensor = torch.tensor([1., 2., 3.])
# 将tensor转为list,再将每个元素转为float类型
float_list = [float(x) for x in tensor.tolist()]
print(float_list)
```
输出:
```
[1., 2., 3.]
```
阅读全文