numpy数据转换成torch类型
时间: 2023-10-25 19:06:54 浏览: 115
Pyorch之numpy与torch之间相互转换方式
可以使用PyTorch中的`torch.from_numpy()`函数将NumPy数组转换为PyTorch中的Tensor类型。示例代码如下:
```python
import numpy as np
import torch
# 创建一个numpy数组
np_array = np.array([1, 2, 3])
# 将numpy数组转换为torch tensor
torch_tensor = torch.from_numpy(np_array)
print(torch_tensor)
```
输出结果为:
```
tensor([1, 2, 3])
```
需要注意的是,`torch.from_numpy()`函数仅在NumPy数组和Tensor共享相同的内存时才会返回一个视图,否则会进行数据复制,这可能会影响性能。
阅读全文