np.array转tensor
时间: 2023-10-18 19:04:29 浏览: 110
浅谈numpy中np.array()与np.asarray的区别以及.tolist
要将一个NumPy数组(`np.array`)转换为PyTorch张量(`torch.Tensor`),可以使用`torch.from_numpy()`函数。下面是一个示例代码:
```python
import numpy as np
import torch
# 创建一个NumPy数组
numpy_array = np.array([[1, 2, 3], [4, 5, 6]])
# 将NumPy数组转换为PyTorch张量
tensor = torch.from_numpy(numpy_array)
print(tensor)
```
输出结果将是一个PyTorch张量:
```
tensor([[1, 2, 3],
[4, 5, 6]])
```
通过这种方式,你可以将NumPy数组转换为PyTorch张量,从而在PyTorch中进行进一步的操作和计算。
阅读全文