将tensor转换为numpy类型
时间: 2023-10-19 14:13:18 浏览: 100
tensor与numpy转换资源合集
5星 · 资源好评率100%
可以使用`numpy()`方法将PyTorch中的tensor转换为NumPy类型。例如,假设有一个名为`my_tensor`的PyTorch张量,可以使用以下代码将其转换为NumPy类型:
```python
import numpy as np
import torch
my_tensor = torch.randn(3, 4)
my_numpy_array = my_tensor.numpy()
```
现在,`my_numpy_array`是一个NumPy数组,可以像普通的NumPy数组一样进行操作。注意,这将创建一个新的NumPy数组,而不是共享内存。如果对NumPy数组进行更改,不会对原始PyTorch张量产生影响。
阅读全文