参考pytorch官方教程,撸pytorch基础知识,记录学习的过程
主要参考DEEP LEARNING WITH PYTORCH: A 60 MINUTE BLITZ,再加上一些numpy的基础知识,这算是一篇对pytorch和numpy的基础知识的总结吧,共分5个方面,这里主要记录1,2部分的学习,英语较好的同学可以直接看参考[1]。
pytorch是基于python的一个科学计算包,主要面向两类同学:
(1)替代numpy从而可以使用GPU来进行计算
(2)作为深度学习平台,提供开发灵活性和速度
1.pytorch基础运算
生成torch数组
torch类型,维度
torch运算(加减)
from __future__ import print_function
import torch
import numpy as np
x_zero = torch.zeros(5, 3, dtype = torch.long)
x_rand2 = torch.rand(4, 5, dtype = torch.double)
x_rand3 = torch.rand(4, 5, dtype = torch.float)
x_rand_add1 = x_rand1 + x_rand3 #同类型才能相加
x_rand_add2 = torch.add(x_rand1, x_rand3) #torch提供的加法
print(x_rand1.size()) #torch.Size([4,