mindspore.ops.Orgqr

class mindspore.ops.Orgqr[源代码]

计算 Householder 矩阵乘积的前 \(N\) 列。

警告

这是一个实验性API,后续可能修改或删除。

更多细节请参考 mindspore.ops.orgqr()

输入:
  • x (Tensor) - shape \((*, M, N)\) 的Tensor,表示二维或者三维矩阵。数据类型为float32、float64、complex64或者complex128。

  • tau (Tensor) - Householder转换的反射系数,其shape为 \((*, K)\) ,其中 K 小于等于 N 。数据类型与 x 一致。

输出:

Tensor,数据类型与shape与 input 一致。

支持平台:

Ascend GPU CPU

样例:

>>> x = Tensor(np.array([[-114.6, 10.9, 1.1], [-0.304, 38.07, 69.38], [-0.45, -0.17, 62.]]), mindspore.float32)
>>> tau = Tensor(np.array([1.55, 1.94, 0.0]), mindspore.float32)
>>> net = ops.Orgqr()
>>> y = net(x, tau)
>>> print(y)
[[-0.54999995 -0.2128925   0.8137956 ]
 [ 0.47119996 -0.8752807   0.08240613]
 [ 0.69749993  0.42560163  0.57772595]]