Pytorch的Tensor和Variable

通过本篇文章，可学习到如何像使用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()