code: def argmax(input, axis=None):
      return torch.argmax(input, dim=axis)
  
imports: from packaging.version import Version
  import numpy
  from .core import Backend, backend_types, backend_basic_math, backend_array
label_desc: Use image node_id 21 for calling the PyTorchBackend obj's underlying member method code with example usage: obj.argmax(input, axis) and returns: torch