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]]