tf.linalg.qr | TensorFlow v2.16.1 (original) (raw)

tf.linalg.qr

Stay organized with collections Save and categorize content based on your preferences.

Computes the QR decompositions of one or more matrices.

View aliases

Compat aliases for migration

tf.linalg.qr(
    input: Annotated[Any, TV_Qr_T], full_matrices: bool = False, name=None
)

Computes the QR decomposition of each inner matrix in tensor such thattensor[..., :, :] = q[..., :, :] * r[..., :,:])

Currently, the gradient for the QR decomposition is well-defined only when the first P columns of the inner matrix are linearly independent, whereP is the minimum of M and N, the 2 inner-most dimmensions of tensor.

# a is a tensor.
# q is a tensor of orthonormal matrices.
# r is a tensor of upper triangular matrices.
q, r = qr(a)
q_full, r_full = qr(a, full_matrices=True)

Args
input	A Tensor. Must be one of the following types: float64, float32, half, complex64, complex128. A tensor of shape [..., M, N] whose inner-most 2 dimensions form matrices of size [M, N]. Let P be the minimum of M and N.
full_matrices	An optional bool. Defaults to False. If true, compute full-sized q and r. If false (the default), compute only the leading P columns of q.
name	A name for the operation (optional).

Returns
A tuple of Tensor objects (q, r).
q	A Tensor. Has the same type as input.
r	A Tensor. Has the same type as input.