5:张量的基本运算

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。

 

posted @ 2021-07-10 16:48  收购阿里巴巴  阅读(305)  评论(0编辑  收藏  举报