code: def print_max_diff_elem(self, ref, tst):
      ref, tst = ref.flatten(), tst.flatten()
      diff = (ref - tst).abs().max()
      idx = (ref - tst).abs().argmax()
      print(
          "Max atol idx: {}, diff: {:.6f}, ref: {:.6f}, tst: {:.6f}".format(
              idx, diff, ref[idx], tst[idx]
          )
      )
  
file_path: apex/apex/contrib/test/xentropy/test_label_smoothing.py
imports: import unittest
  import random
  import time
  import numpy
  import torch
label_desc: Use image node_id 3 for calling the LabelSmoothingTest obj's underlying member method code with example usage: obj.print_max_diff_elem(ref, tst) without return types