如何将float类型tensor转换为double型tensor张量
时间: 2023-11-30 16:02:26 浏览: 80
PyTorch中Tensor的数据类型和运算的使用
你可以使用 PyTorch 中的 `.double()` 方法将 float 类型的 tensor 转换为 double 类型的 tensor,例如:
```python
import torch
# 创建一个 float 类型的 tensor
float_tensor = torch.tensor([1.0, 2.0, 3.0])
# 将 float 类型的 tensor 转换为 double 类型的 tensor
double_tensor = float_tensor.double()
# 打印结果
print(double_tensor)
```
输出结果为:
```
tensor([1., 2., 3.], dtype=torch.float64)
```
阅读全文