QR-разложение в TensorFlow

Question

Я видел, что существуют методы для разложения Холецкого и решения линейных систем с использованием метода QR в TensorFlow, однако я не могу найти способ реализовать разложение QR в TensorFlow.

Как вы выполняете QR-разложение в TensorFlow?

Answer 1

Кажется, что QR-разложение сейчас находится в ведении tenorflow; это было добавлено на прошлой неделе.

tf.qr(input, full_matrices=None, name=None)

Вычисляет QR-разложения одной или нескольких матриц.

Вычисляет QR-разложение каждой внутренней матрицы в tensor такой, что tensor[..., :, :] = q[..., :, :] * r[..., :,:])

# 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)