怎么将CUDA:0设备类型的张量转换为numpy。
时间: 2024-02-23 10:00:55 浏览: 62
PyTorch常用代码段合集1
您可以使用 `.cpu().numpy()` 方法将CUDA设备上的张量转换为NumPy数组,例如:
```
import torch
# 创建一个CUDA设备上的张量
x = torch.randn(2, 3).cuda()
# 将张量转换为NumPy数组
x_np = x.cpu().numpy()
print(x_np)
```
输出:
```
[[-0.06586416 -0.16125001 -0.15470208]
[-0.14527392 -0.15236327 0.8335287 ]]
```
阅读全文