zoukankan      html  css  js  c++  java
  • 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。

  • 相关阅读:
    逻辑回归问题(Logistic Regression)
    丑数
    用两个栈实现队列
    重建二叉树
    单变量的线性回归(Linear Regression with One Variable)
    机器学习
    二维数组中的查找
    面经
    亚信实习---->PLSQL链接Oracle
    亚信实习小练习
  • 原文地址:https://www.cnblogs.com/jiafeng1996/p/14994099.html
Copyright © 2011-2022 走看看