首页 > 其他 > 详细

5:张量的基本运算

时间:2021-07-10 21:59:10      阅读:25      评论:0      收藏:0      [点我收藏+]

1:基本运算符(重载的+-*/或者函数add(a,b),sub(a,b),mul(a,b),div(a,b))实现基本的加减乘除。

技术分享图片

 

 【注】可以用eq()函数查看两种不同运算之后的维度信息是否相同,all()函数可以查看对用维度的数据元素是否相同。

2:matmul/mm/@都是矩阵相乘

技术分享图片

 

 【注】mm仅仅适用于二维矩阵

技术分享图片

 

 [注]大于二维的tensor,用matmul进行矩阵相乘:使用的机制是取后两维进行2维的矩阵运算,前面的维度保持不变;如果

当维度不相等时会使用broadcasting机制进行维度的扩张。

3:power()/**次方运算

技术分享图片

 

4:exp()/log()这里的log默认以e为底

 技术分享图片

 5:近似值

技术分享图片

 

 

 【注】.trunc()裁剪小数部分的整数部分,.frac()裁剪小数部分的小数部分,.ceil()向上取整,.floor()向下取整。round()函数四舍五入

6:clamp()函数:梯度裁剪

技术分享图片

 

 【注】当clamp(min)含有一个参数时:则是所有元素最小为min,如果小于min,则将其置为min。

clamp(min,max)含有两个参数时:则是所有元素最小为min,最大为max。如果有值小于min,则将其置为min;如果有值大于max,则将其置为max。

 

5:张量的基本运算

原文:https://www.cnblogs.com/jiafeng1996/p/14994099.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!