torch.from_numpy具体用法
时间: 2023-05-17 18:03:38 浏览: 387
Pyorch之numpy与torch之间相互转换方式
torch.from_numpy可以将numpy数组转换为PyTorch张量。具体用法如下:
```python
import numpy as np
import torch
# 创建一个numpy数组
arr = np.array([1, 2, 3])
# 将numpy数组转换为PyTorch张量
tensor = torch.from_numpy(arr)
print(tensor)
```
输出结果为:
```
tensor([1, 2, 3])
```
需要注意的是,torch.from_numpy返回的张量与原始的numpy数组共享内存,因此对张量的修改也会影响到原始的numpy数组。如果需要避免这种情况,可以使用torch.Tensor(arr)创建一个新的张量。
阅读全文