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

相关文章:

  • 2021-06-13
  • 2022-12-23
  • 2021-10-05
  • 2021-06-07
  • 2022-02-19
  • 2021-12-04
  • 2021-07-21
  • 2022-01-17
猜你喜欢
  • 2021-07-18
  • 2022-12-23
  • 2021-11-04
  • 2021-12-02
  • 2021-05-09
  • 2021-12-24
  • 2022-01-11
相关资源
相似解决方案