1、torch.mul(a, b)是矩阵a和b对应位相乘,a和b的维度必须相等,比如a的维度是(1, 2),b的维度是(1, 2),返回的仍是(1, 2)的矩阵;
2、torch.mm(a, b)是矩阵a和b矩阵相乘,比如a的维度是(1, 2),b的维度是(2, 3),返回的就是(1, 3)的矩阵。
3 torch.nn.Parameter 这个属于在神经网络中可训练参数矩阵的初始化中使用。表示整个矩阵融入模型中并且随着模型的训练不断地改变。
4 repeat() repeat(1,2)=(1,1)
5 torch.cat 拼接矩阵用的,不多说了
6 view()torch中的改变矩阵形状的(-1,xx)-1表示矩阵的形状需要随机应变的时候,让电脑帮忙计算
7 ones_like ,zeros_like 生成一个参数矩阵同形状的矩阵全为1/全为0
Pytorch学习笔记【自用】
最新推荐文章于 2024-08-14 09:34:07 发布