Pytorch tensor
0. 环境介绍
环境使用 Kaggle 里免费建立的 Notebook
引入 pytorch
import torch
1. tensor
1.1 基本操作
张量(tensor)数值组成的数组,数组可以有多个维度
1.1.1 arange()
生成向量
1.1.2 shape
输出张量形状
1.1.3 numel()
返回张量中元素总数
1.1.4 reshape()
改变一个张量的形状
reshape()
适用于元素数量相同的所有张量
1.1.5 zeros()
全 0 阵
1.1.6 ones()
全 1 阵
1.1.7 tensor
列表赋值
1.2 tensor 运算
1.2.1 常见的标准算术运算符
常见的标准算术运算符(+
,-
,*
,/
,**
)都是按照对应元素运算,**
是求幂运算
1.2.2 exp()
指数运算
1.2.3 cat()
连接张量,dim 参数值默认为 0(下面拼接),为 1 时表示右边拼接
1.2.4 ==
逻辑运算
1.2.5 sum()
对张量内元素求和求和
1.2.6 广播机制
元素操作,b 张量将 0,1 分别进行列扩展再对应元素运算
1.2.7 张量访问
1.3 执行原地操作
运行一些操作可能会导致新结果分配内存
执行原地操作不会分配新的内存,zeros_like
表示生成一个与 Y 张量 shape 相同的全 0 阵