Computes the gradient function for function f via backpropagation.
tf.raw_ops.SymbolicGradient(
input, Tout, f, name=None
)
| Args | |
|---|---|
input | A list of Tensor objects. a list of input tensors of size N + M; |
Tout | A list of tf.DTypes that has length >= 1. the type list for the input list. |
f | A function decorated with @Defun. The function we want to compute the gradient for. The function 'f' must be a numerical function which takes N inputs and produces M outputs. Its gradient function 'g', which is computed by this SymbolicGradient op is a function taking N + M inputs and produces N outputs. I.e. if we have (y1, y2, ..., y_M) = f(x1, x2, ..., x_N), then, g is (dL/dx1, dL/dx2, ..., dL/dx_N) = g(x1, x2, ..., x_N, dL/dy1, dL/dy2, ..., dL/dy_M), where L is a scalar-value function of (x1, x2, ..., xN) (e.g., the loss function). dL/dx_i is the partial derivative of L with respect to x_i. (Needs some math expert to say the comment above better.) |
name | A name for the operation (optional). |
| Returns | |
|---|---|
A list of Tensor objects of type Tout. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/SymbolicGradient