GitHub

tf_agents.bandits.policies.linalg.conjugate_gradient_solve

Returns X such that A * X = B.

tf_agents.bandits.policies.linalg.conjugate_gradient_solve(
    a_mat: tf_agents.typing.types.Tensor,
    b_mat: tf_agents.typing.types.Tensor,
    tol: float = 1e-10
) -> tf_agents.typing.types.Tensor

Uses Conjugate Gradient to solve many linear systems of equations with the same matrix a_mat and multiple right hand sides provided as columns in the matrix b_mat.

Args
`a_mat`	a Symmetric Positive Definite matrix, represented as a `Tensor` of shape `[n, n]`.
`b_mat`	a `Tensor` of shape `[n, k]`.
`tol`	(float) desired tolerance on the residual.

Returns
`X`	`Tensor` `X` of shape `[n, k]` such that `A * X = B`.

Raises
`ValueError`	if `a_mat` is not square or `a_mat` and `b_mat` have incompatible shapes.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2021-01-20 UTC.

tf_agents.bandits.policies.linalg.conjugate_gradient_solve

Args

Returns

Raises