code: def predict_from_logits(logits, dim=1):
      return logits.max(dim=dim, keepdim=False)[1]
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from __future__ import unicode_literals
  import torch
  import torch.nn
  import torch.nn.functional
label_desc: Use image node_id 27 for calling a global function with example usage: predict_from_logits(logits, dim) and returns: logits