tensor类型转换ndarray
时间: 2023-09-25 21:15:50 浏览: 91
详解pytorch tensor和ndarray转换相关总结
5星 · 资源好评率100%
可以使用numpy中的array函数将tensor类型转换为ndarray,例如:
```python
import torch
import numpy as np
tensor = torch.ones((2, 3))
ndarray = np.array(tensor)
```
这样就可以将大小为2x3的全1张量转换为ndarray类型的数据。
阅读全文