code: def argmax(self, dim=None, keepdim=False, one_hot=True):
      """Returns the indices of the maximum value of all elements in the
      `input` tensor.
      """
      method = cfg.functions.max_method
  
      if self.dim() == 0:
          result = (
              self.new(torch.ones((), device=self.device))
              if one_hot
              else self.new(torch.zeros((), device=self.device))
          )
          return result
  
      result = _argmax_helper(
          self, dim, one_hot, method, _return_max=False
      )
  
      if not one_hot:
          result = _one_hot_to_index(result, dim, keepdim, self.device)
      return result
  
file_path: CrypTen/crypten/common/functions/maximum.py
imports: import math
  import crypten
  import torch
  from crypten.config import cfg
label_desc: Use image node_id 1 for calling a global function with example usage: argmax(dim, keepdim, one_hot) and returns: result, result