+:add或者+
-:sub或者-*:mul或者*
/:div或者/——一个**/符号代表除法,两个//**代表整除
矩阵相乘*:是对应位置相乘
torch.mm只适用2d,不推荐
[email protected]:矩阵相乘,@是matmul的重载
图中w为(512,784),a(4,784)@w(784,512)=(4,512),为什么w的512放第一位呢:Pytorch习惯把输出放第一位,t():转置,适合2d,高维用transpose
四维*四维:前两位不变,矩阵乘后两位,前两位如果不一致就适用broadcast
次方函数
sqrt:平方根
rsqrt:倒数
e,log
以2为底:log2
以10为底:log10
floor(地板):往下走取整
ceil(天花板):往上走取整
trunc,frac(裁剪):裁剪为整数部分和小数部分
round:四舍五入
clamp(裁剪)
clamp(10):裁剪到最小值为10
clamp(0,10):裁剪到0-10
相关文章: