tf.recompute_grad
| View source on GitHub |
An eager-compatible version of recompute_grad.
tf.recompute_grad(
f
)
For f(*args, **kwargs), this supports gradients with respect to args or kwargs, but kwargs are currently only supported in eager-mode. Note that for keras layer and model objects, this is handled automatically.
| Args | |
|---|---|
f | function f(*x) that returns a Tensor or sequence of Tensor outputs. |
| Returns | |
|---|---|
A function g that wraps f, but which recomputes f on the backwards pass of a gradient call. |
© 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/recompute_grad
