通过本篇文章,可学习到如何像使用numpy一样使用Pytorch,了解到Pytorch的基本元素tensor和variable。
把pytorch当作numpy使用
下面展示一些 具体代码片
。
import torch
import numpy as np
#创建一个numpy ndarray
numpy_tensor = np.random.randn(10, 20)
我们可以使用下面两种方式将numpy的ndarray转换到tensor上
pytorch_tensor1 = torch.Tensor(numpy_tensor)
pytorch_tensor2 = torch.from_numpy(numpy_tensor)
我们也可以使用下面方式将pytorch tensor 转换为numpy的ndarray
# 如果tensor在gpu上
numpy_array = pytorch_tensor1.numpy()
# 如果tensor在cpu上
numpy_array = pytorch_tensor1.cpu().numpy()