mindspore.ops.ormqr

查看源文件
mindspore.ops.ormqr(input, tau, other, left=True, transpose=False)[源代码]

计算一个普通矩阵 other 与一个矩阵Q(由Householder反射向量 input 和反射系数 tau 生成)的乘积。

如果 leftTrue ,计算顺序为Q * other ,否则,计算顺序为 other * Q。

参数:
  • input (Tensor) - 输入tensor,shape (,mn,k),当 leftTrue 时, mn的值等于m,否则mn的值等于n。

  • tau (Tensor) - 输入tensor,shape (,min(mn,k))

  • other (Tensor) - 输入tensor,shape (,m,n)

  • left (bool, 可选) - 计算顺序。默认 True

  • transpose (bool, 可选) - 是否对矩阵Q进行共轭转置变换。默认 False

返回:

Tensor

支持平台:

GPU

样例:

>>> import mindspore
>>> input = mindspore.tensor(([[-114.6, 10.9, 1.1], [-0.304, 38.07, 69.38], [-0.45, -0.17, 62]]),
...                             mindspore.float32)
>>> tau = mindspore.tensor(([1.55, 1.94, 3.0]), mindspore.float32)
>>> other = mindspore.tensor(([[-114.6, 10.9, 1.1],
...                            [-0.304, 38.07, 69.38],
...                            [-0.45, -0.17, 62]]), mindspore.float32)
>>> output = mindspore.ops.ormqr(input, tau, other)
>>> print(output)
[[  63.82713   -13.823125 -116.28614 ]
 [ -53.659264  -28.157839  -70.42702 ]
 [ -79.54292    24.00183   -41.34253 ]]