torch.from_numpy(np.array())
时间: 2024-01-21 19:04:05 浏览: 153
Pyorch之numpy与torch之间相互转换方式
这个函数是将一个 NumPy 数组转换为 PyTorch 张量。它的作用是在 PyTorch 和 NumPy 之间实现数据的无缝转换。具体来说,它将 NumPy 数组中的数据类型转换为 PyTorch 张量支持的数据类型,并且共享底层内存,因此在转换时不需要复制数据,从而提高了效率。用法示例:
```python
import numpy as np
import torch
# 创建一个 NumPy 数组
arr = np.array([[1, 2, 3], [4, 5, 6]])
# 将 NumPy 数组转换为 PyTorch 张量
tensor = torch.from_numpy(arr)
print(tensor)
# 输出:
# tensor([[1, 2, 3],
# [4, 5, 6]], dtype=torch.int32)
```
阅读全文