tf.raw_ops.Cholesky

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

tf.raw_ops.Cholesky(
    input, name=None
)

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

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

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

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