首页怎么将CUDA:0设备类型的张量转换为numpy。

怎么将CUDA:0设备类型的张量转换为numpy。

时间: 2024-02-23 16:00:55 浏览: 57

pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换

1, 创建pytorch 的Tensor张量： torch.rand((3,224,224)) #创建随机值的三维张量，大小为（3,224,224） torch.Tensor([3,2]) #创建张量，[3,2] 2, cpu上的tensor和GPU即pytorch创建的tensor的相互转化 b = a.cpu() # GPU → CPU a = b.cuda() #CPU → GPU 3, tensor和numpy的转化 b = a.numpy() # tensor转化为 numpy数组 a = b.from_numpy() # numpy数组转化为tensor 4, torch的

您可以使用 `.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 ]] ```

阅读全文