First public release - v1.0.0!
Summary
tensorflow-hooks
(importable as tf_hooks
) provides PyTorch-like hooks for TensorFlow Keras layers. It provides two main functions:
register_forward_hook
for forward hooks, which execute after forward inference of the layer.register_forward_pre_hook
for forward pre-hooks, which execute prior to forward inference of the layer.
See the README for more details.