Pytorch基础学习

Tensor

import torch
x1 = torch.Tensor([3,4])
x2 = torch.FloatTensor([4,4])
x3 = torch.randn(3,4)

pytorch与numpy之间的转换

import numpy as np
numpy_tensor = np.random.randn(10,20)
pytorch_tensor = torch.from_numpy(numpy_tensor)  #numpy转换pytorch
new_numpy_tensor = pytorch_tensor.numpy()  #pytorch转numpy

GPU tensor和CPU tensor之间的转换

x = torch.randn(3,4)
x_gpu = x.cuda(0)
x_gpu = x.cuda(1)
x_cpu = x_gpu.cpu()
x_array = x_gpu.cpu().numpy()   #先转换CPU再转换成numpy

两种方式将Tensor放到GPU上

#第一种：定义cuda数据类型
dtype = torch.cuda.FloatTensor   #定义默认GPU的数据类型
gpu_tensor = torch.randn(10,20).type(dtype)

#第二种：简单推荐使用
gpu_tensor = torch.randn(10,20).cuda(0)  #将tensor放到第一个GPU上
gpu_tensor = torch.randn(10,20).cu