tf.linalg.cholesky

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

tf.linalg.cholesky(
    input, name=None
)

Ввод — это тензор формы [..., M, M], внутренние 2 измерения которого образуют квадратные матрицы.

Ввод должен быть симметричным и положительно определённым. Для этой операции будет использоваться только нижняя треугольная часть ввода. Верхняя треугольная часть не будет читаться.

Вывод — тензор той же формы, что и входной, содержащий разложения Холецкого для всех входных подматриц [..., :, :].

Примечание: Вычисление градиента на GPU быстрее для больших матриц, но не для больших размерностей пакетов, когда подматрицы небольшие. В этом случае может быть быстрее использовать CPU.