我的第一次pytorch 基础练习

一、pytorch 基础

1、定义数据

 

 

2、定义张量

 张量 tensor，是数字各种形式的总称。

 

 

 

 3、定义操作

凡是用Tensor进行各种运算的，都是Function

• 基本运算，加减乘除，求幂求余
  • abs/sqrt/div/exp/fmod/pow ，及一些三角函数 cos/ sin/ asin/ atan2/ cosh，及 ceil/round/floor/trunc
• 布尔运算，大于小于，最大最小
  • gt/lt/ge/le/eq/ne，topk, sort, max/min
• 线性运算，矩阵乘法，求模，求行列式
  • trace, diag, mm/bmm，t，dot/cross，inverse，svd 等

 

 

 

 

 

 

 

 

 

 

 

二、螺旋数据分类

1、准备工作

下载绘图函数到本地，引入基本的库，然后初始化重要参数，初始化 X 和 Y, 其中 X 是特征矩阵，Y是样本标签。

 

 

 

 

 

2、构建线性模型分类

 

 

 

3、构建两层神经网络分类