def trace(a, offset=0, axis1=0, axis2=1, dtype=None): t = _linalg.trace(a, offset) dtype = _translatenativetype(dtype) if dtype is not None: t = t.cast(dtype.value) return t
def trace(a, offset=0, axis1=0, axis2=1, dtype=None): dtype = _translatenativetype(dtype) if dtype is None: dtval = a.getDType() else: dtval = dtype.value return _linalg.trace(a, offset).cast(dtval)