基础

加减法

遇到不匹配情况,会自动进行broadcasting进行维度匹配后,再进行操作

运算符得到了重载

tensor运算符

乘法

Torch.mm():仅在2维中使用

Torch.matmul()

@重载后的乘法操作符

tensor运算符

即使是多维矩阵相乘,实际相乘的仍然是最后两维数据,所以前后行列向量维数必须匹配

当其他维度不匹配时,仍然使用Broadingcast进行维度匹配后运算

tensor运算符

Power&sqrt

.rsqrt():平方根的倒数

tensor运算符

对数、指数

tensor运算符

近似值

floor():向下取整

ceil():向上取整

trunc():截取整数位

frac():截取小数位

round():四舍五入

tensor运算符

clamp

一个参数,参数为最小值

两个参数,表示(最小值,最大值)

tensor运算符

 

 

相关文章:

  • 2022-02-09
  • 2022-12-23
  • 2022-12-23
  • 2021-06-01
  • 2021-09-14
  • 2021-09-25
  • 2022-12-23
  • 2021-07-17
猜你喜欢
  • 2021-07-13
  • 2021-11-27
  • 2021-08-24
  • 2022-12-23
  • 2022-12-23
  • 2021-12-03
  • 2022-12-23
相关资源
相似解决方案