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)