mindspore.ops.geqrf

查看源文件
mindspore.ops.geqrf(input)[源代码]

将输入tensor进行QR分解 \(A = QR\)

分解为正交矩阵 Q 和上三角矩阵 R 的乘积。

警告

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

参数:
  • input (Tensor) - 输入tensor。

返回:

两个tensor组成的tuple( y , tau )。

  • y (Tensor) - 隐式存储 QR 矩阵。 Q (Householder反射向量)存储在对角线下方, R 的元素存储在对角线及上方。

  • tau (Tensor) - 存储每个Householder变换的缩放因子(Householder反射系数)。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> input = mindspore.tensor([[-2.0, -1.0], [1.0, 2.0]])
>>> y, tau = mindspore.ops.geqrf(input)
>>> print(y)
[[ 2.236068   1.7888544]
 [-0.236068   1.3416407]]
>>> print(tau)
[1.8944271 0.       ]