torch.tensor与numpy相互转换
时间: 2024-05-14 08:19:48 浏览: 104
pytorch 实现tensor与numpy数组转换
可以使用以下两种方法进行torch.tensor与numpy相互转换:
1. 将numpy数组转换为torch.tensor:
```
import numpy as np
import torch
# 创建一个numpy数组
arr = np.array([1, 2, 3])
# 将numpy数组转换为torch.tensor
tensor = torch.from_numpy(arr)
```
2. 将torch.tensor转换为numpy数组:
```
import numpy as np
import torch
# 创建一个torch.tensor
tensor = torch.tensor([1, 2, 3])
# 将torch.tensor转换为numpy数组
arr = tensor.numpy()
```
阅读全文