tf.linalg.cholesky

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

tf.linalg.cholesky(
    input: Annotated[Any, TV_Cholesky_T], name=None
) -> Annotated[Any, TV_Cholesky_T]

Используется в блокнотах

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

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

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

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