docstr: Returns the index of the maximum value in a tensor.
docstr_tok: ["Returns","the","index","of","the","maximum","value","in","a","tensor","."]
code: def argmax(X: tf.Tensor, dim=-1) -> tf.Tensor:
      """Returns the index of the maximum value in a tensor."""
      X = tf.math.argmax(X, axis=dim)
      return X
  
file_path: alibi/alibi/explainers/similarity/backends/tensorflow/base.py
imports: from typing import Any, Callable, List, Optional, Union
  import numpy
  import tensorflow
  from tensorflow import keras
label_desc: Use image node_id 6 for calling the _TensorFlowBackend obj's underlying member method code with example usage: obj.argmax(X, dim) and returns: X