tf.linalg.cholesky

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

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

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

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

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

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