-
加减乘除
直接用 +, -, * ,/
或者torch.add, torch.sub, torch.mul, torch.div
效果等同 -
矩阵乘法
前面的mul和*是元素相乘
矩阵乘法用 torch.matmul或者 重载的@

超过2维的矩阵相乘,是用最后两维进行矩阵乘法,例如:

如果是(4,3,28,32)与(4,1,64,32),则会先做广播(broadcast)处理,再做矩阵乘法。(但必须符合可以广播的规则) -
power
torch.power 或者 ** -
sqrt (平方根),rsqrt (平方根再倒数)
-
Exp(e的x次方), log(e为底的对数)

-
近似计算

floor: 下取整, ceil:上取整,trunc:整数部分 , frac:小数部分,round:四舍五入取整
pytorch学习笔记7 tensor基本数学运算
最新推荐文章于 2025-12-04 16:10:29 发布

最低0.47元/天 解锁文章
1870

被折叠的 条评论
为什么被折叠?



