将numpy转换为dtype=float32的torch张量
时间: 2024-01-31 20:03:49 浏览: 136
Pyorch之numpy与torch之间相互转换方式
可以使用 `torch.from_numpy()` 方法将 numpy 数组转换为 PyTorch 张量,并通过 `torch.Tensor()` 方法指定数据类型为 `torch.float32`,示例如下:
```python
import numpy as np
import torch
# 创建一个 numpy 数组
arr = np.array([1, 2, 3])
# 将 numpy 数组转换为 PyTorch 张量,并指定数据类型为 float32
tensor = torch.from_numpy(arr).type(torch.Tensor).to(torch.float32)
```
在上面的示例中,我们首先创建了一个 numpy 数组 `arr`,然后使用 `torch.from_numpy()` 方法将其转换为 PyTorch 张量。接着,使用 `torch.Tensor()` 方法指定了张量的数据类型为 `torch.float32`,最后使用 `.to()` 方法将张量移动到指定的设备上(这里是 CPU)。
阅读全文