将numpy数组转化为pytorch的张量的代码
时间: 2023-06-05 19:05:56 浏览: 117
pytorch 实现tensor与numpy数组转换
可以使用以下代码将numpy数组转换为pytorch的张量:
```
import torch
import numpy as np
# 创建一个numpy数组
arr = np.array([1, 2, 3, 4])
# 将numpy数组转为pytorch张量
tensor = torch.from_numpy(arr)
print(tensor)
```
阅读全文