class_docstr: Draw activations of a specific layer for a given input
docstr: Draw activations outputs at each epoch end to Tensorboard.
  
  Args:
      epoch (int): Epoch index
      logs (dict): Additional information on epoch
code: def on_epoch_end(self, epoch, logs=None):
      """
      Draw activations outputs at each epoch end to Tensorboard.
  
      Args:
          epoch (int): Epoch index
          logs (dict): Additional information on epoch
      """
      explainer = ExtractActivations()
      grid = explainer.explain(
          self.validation_data, self.model, self.layers_name
      )
  
      # Using the file writer, log the reshaped image.
      with self.file_writer.as_default():
          tf.summary.image(
              "Activations Visualization",
              np.array([np.expand_dims(grid, axis=-1)]),
              step=epoch,
          )
  
file_path: tf-explain/tf_explain/callbacks/activations_visualization.py
imports: from datetime import datetime
  from pathlib import Path
  import numpy
  import tensorflow
  from tensorflow.keras.callbacks import Callback
  from tf_explain.core.activations import ExtractActivations
label_desc: Use image node_id 2 for calling the ActivationsVisualizationCallback obj's underlying member method code with example usage: obj.on_epoch_end(epoch, logs) without return types