torch numpy转张量
时间: 2023-11-12 09:06:53 浏览: 88
Pytorch使用PIL和Numpy将单张图片转为Pytorch张量方式
可以使用`torch.from_numpy()`函数将numpy数组转换为PyTorch张量。
示例代码:
```
import numpy as np
import torch
# 创建一个numpy数组
arr = np.array([[1, 2, 3], [4, 5, 6]])
# 将numpy数组转换为PyTorch张量
tensor = torch.from_numpy(arr)
print(tensor)
```
输出:
```
tensor([[1, 2, 3],
[4, 5, 6]], dtype=torch.int32)
```
阅读全文