将tensor转化为ndarray格式
时间: 2023-08-12 07:06:36 浏览: 89
详解pytorch tensor和ndarray转换相关总结
5星 · 资源好评率100%
在PyTorch中,可以使用`numpy()`方法将Tensor转换为NumPy数组。具体地,可以使用以下代码将一个Tensor对象`t`转换为NumPy数组:
```python
import torch
t = torch.tensor([1, 2, 3])
arr = t.numpy()
```
这里,我们首先定义了一个Tensor对象`t`,然后使用`t.numpy()`方法将其转换为NumPy数组`arr`。需要注意的是,该方法会返回一个和原Tensor共享数据存储空间的NumPy数组,因此修改NumPy数组的值也会影响到原Tensor。
如果想要将CPU上的Tensor转换为NumPy数组,该方法可以直接使用。如果Tensor在GPU上,需要先使用`cpu()`方法将其转移到CPU上,再使用`numpy()`方法将其转换为NumPy数组:
```python
import torch
t = torch.tensor([1, 2, 3]).cuda()
arr = t.cpu().numpy()
```
这里,我们首先将Tensor对象`t`移动到GPU上,然后使用`t.cpu().numpy()`方法将其转换为NumPy数组。
阅读全文