code: def trace(a):
      return _np.trace(a, axis1=-2, axis2=-1)
  
imports: from ._dispatch import BACKEND_NAME, _common
  from ._dispatch import numpy
label_desc: Use image node_id 28 for calling a global function with example usage: trace(a) and returns: _np