tf.keras.backend.gradients

View source on GitHub

Returns the gradients of loss w.r.t. variables.

tf.keras.backend.gradients(
    loss, variables
)

Arguments:

• loss: Scalar tensor to minimize.
• variables: List of variables.

Returns:

A gradients tensor.