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: torch.Tensor, dim=-1) -> torch.Tensor:
      """Returns the index of the maximum value in a tensor."""
      return torch.argmax(X, dim=dim)
  
file_path: alibi/alibi/explainers/similarity/backends/pytorch/base.py
imports: from typing import Any, Callable, List, Optional, Union
  import numpy
  import torch
  import torch.nn
label_desc: Use image node_id 6 for calling the _PytorchBackend obj's underlying member method code with example usage: obj.argmax(X, dim) and returns: torch