mindspore.ops.orgqr
- mindspore.ops.orgqr(input, input2)[源代码]
计算Householder矩阵的前
列。通常用于计算
mindspore.ops.Geqrf
返回的正交矩阵 的显式表示。下面以输入无batch维的情况为例: 假设输入 input 的shape经过Householder转换之后为:
。 当 input 的对角线被置为1, input 中下三角形的每一列都表示为: ,其中 在 范围内,此函数返回Householder矩阵乘积的前 列:其中:
是 维单位矩阵。当 是复数的时候, 是共轭转置,否则是一般转置。输出的shape与输入shape相同。 即输入 input2 。- 参数:
input (Tensor) - 二维或三维输入tensor,表示Householder反射向量, shape
。input2 (Tensor) - 一维或二维输入tensor,表示Householder反射系数,shape为
,其中 K 小于等于 N 。
- 返回:
Tensor
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> input = mindspore.tensor([[-2.0, -1.0], [1.0, 2.0]]) >>> y, tau = mindspore.ops.geqrf(input) >>> mindspore.ops.orgqr(y, tau) Tensor(shape=[2, 2], dtype=Float32, value= [[-8.94427061e-01, 4.47213590e-01], [ 4.47213590e-01, 8.94427180e-01]])