Pytorch - Variable_Tensor_Numpy_Cuda 之间的转换

最新推荐文章于 2024-01-08 08:39:12 发布

极客Array

最新推荐文章于 2024-01-08 08:39:12 发布

阅读量1.6k

点赞数 1

分类专栏： PyTorch 文章标签： Variable Tensor Numpy Cuda

本文链接：https://blog.csdn.net/Harpoon_fly/article/details/84345572

版权

本文介绍了Pytorch中Variable、Tensor、Numpy间以及CUDA转换的操作，包括Variable转Numpy、Numpy转Variable、Tensor转Numpy、Numpy转Tensor的步骤，并展示了如何将数据复制到GPU进行计算以及从GPU返回CPU的操作，同时指出了从CUDA取数据时避免错误的方法。

摘要由CSDN通过智能技术生成

Pytroch 涉及到 Variable，Tensor 和 Numpy 间的转换比较多，还会涉及到 cuda 到 cpu的转换.

Variable 转 Numpy

 import torch
 from torch.autograd import Variable

var = Variable(torch.FloatTensor(2,3))
# var = tensor(1.00000e-03 *
#       [[ 1.1476,  0.0000,  0.0000],
#        [ 0.0000,  0.0000,  0.0000]])
var_numpy = var.data.numpy()
# array([[1.1476139e-03, 4.5816855e-41, 3.7984453e-37],
#        [0.0000000e+00, 4.4841551e-44, 0.0000000e+00]], dtype=float32)

1.2 Numpy 转 Variable

import torch
from torch.autograd import Variable
import numpy as np

var_numpy = np.random.randn(2, 3)
# array([[-0.27443182,  1.18369008, -0.24645608],
#        [-0.99800364,  0.58202014, -0.84904032]])
var = Variable(torch.from_numpy(var_numpy))
# tensor([[-0.2744,  1.1837, -0.2465],
#         [-0.9980,  0.5820, -0.8490]], dtype=torch.float64)

1.3 Tensor 转 Numpy