矩阵和矢量乘积

矩阵和向量乘积计算包括:

  • 矩阵乘法
  • 向量乘法
  • 计算协方差矩阵
  • 计算两个向量的互相关
  • 计算两个向量的卷积
  • 计算相关系数  

函数

动作

MatMul

两个矩阵的矩阵乘积

GeMM

通用矩阵乘法(GeMM)

Power

将方阵提升整数幂

Dot

两个向量的点积

Kron

返回两个矩阵、矩阵和向量,向量和矩阵、或两个向量的克罗内克(Kronecker)乘积

Inner

两个矩阵的内积

Outer

计算两个矩阵或两个向量的外积

CorrCoef

计算皮尔逊(Pearson)相关系数(线性相关系数)

Cov

计算协方差矩阵

Correlate

计算两个向量的互相关性

Convolve

返回两个向量的离散线性卷积