tf.keras.ops.qr

Computes the QR decomposition of a tensor.

tf.keras.ops.qr(
    x, mode='reduced'
)

x Input tensor of shape (..., M, N).
mode A string specifying the mode of the QR decomposition.

  • 'reduced': Returns the reduced QR decomposition. (default)
  • 'complete': Returns the complete QR decomposition.

A tuple containing two tensors. The first tensor of shape (..., M, K) is the orthogonal matrix q and the second tensor of shape (..., K, N) is the upper triangular matrix r, where K = min(M, N).

Example:

x = keras.ops.convert_to_tensor([[1., 2.], [3., 4.], [5., 6.]])
q, r = qr(x)
print(q)
array([[-0.16903079  0.897085]
       [-0.5070925   0.2760267 ]
       [-0.8451542  -0.34503305]], shape=(3, 2), dtype=float32)