将numpy.nparray数组转化为tensor代码
时间: 2023-11-27 21:49:58 浏览: 102
tensor和numpy的互相转换的实现示例
可以使用PyTorch中的torch.from_numpy()方法将NumPy数组转换为Tensor。以下是示例代码:
```python
import numpy as np
import torch
# 创建一个NumPy数组
np_array = np.array([[1, 2], [3, 4]])
# 将NumPy数组转换为Tensor
tensor = torch.from_numpy(np_array)
print(tensor)
```
输出:
```
tensor([[1, 2],
[3, 4]])
```
请注意,转换后的Tensor与原始的NumPy数组共享内存,因此在更改其中一个时,另一个也会发生变化。如果要避免这种情况,可以使用Tensor的clone()方法创建一个副本。
阅读全文