code: def graymap(X):
      return ivis.graymap(np.abs(X), input_is_positive_only=True)
  
file_path: innvestigate/examples/utils/mnist.py
imports: from __future__ import annotations
  import numpy
  import tensorflow.keras
  from tensorflow.keras.backend import image_data_format
  from tensorflow.keras.datasets import mnist
  import innvestigate.utils
  import innvestigate.utils.visualizations
label_desc: Use image node_id 7 for calling a global function with example usage: graymap(X) and returns: ivis