tensor转ndarray
时间: 2023-11-12 18:08:38 浏览: 98
计算机视觉——【tensorflow入门】Tensor与Numpy.ndarray的相互转换 计算机视觉.pdf
在Python中,可以使用NumPy库将Tensor转换为ndarray。Tensor是PyTorch中的一种数据类型,而NumPy中的ndarray是一种多维数组。
以下是将PyTorch中的Tensor转换为NumPy中的ndarray的示例代码:
```python
import torch
import numpy as np
# 创建一个Tensor
tensor = torch.tensor([[1, 2], [3, 4]])
# 将Tensor转换为ndarray
ndarray = tensor.numpy()
print(ndarray)
# 输出:array([[1, 2],
# [3, 4]])
```
在上面的代码中,我们首先创建了一个包含两个元素的2x2张量。然后,我们使用`numpy()`方法将Tensor转换为ndarray,并将其打印出来。输出与原始Tensor相同,但类型为NumPy的ndarray。
阅读全文