mindspore.ops.ormqr
- mindspore.ops.ormqr(input, tau, other, left=True, transpose=False)[源代码]
计算一个普通矩阵 other 与一个矩阵Q(由Householder反射向量 input 和反射系数 tau 生成)的乘积。
如果 left 为
True
,计算顺序为Q * other ,否则,计算顺序为 other * Q。- 参数:
input (Tensor) - 输入tensor,shape
,当 left 为True
时, mn的值等于m,否则mn的值等于n。tau (Tensor) - 输入tensor,shape
。other (Tensor) - 输入tensor,shape
。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 ]]