Function trace

dask/array/reductions.py:1477–1478

(a, offset=0, axis1=0, axis2=1, dtype=None)

@derived_from(np)
def trace(a, offset=0, axis1=0, axis2=1, dtype=None):
    return diagonal(a, offset=offset, axis1=axis1, axis2=axis2).sum(-1, dtype=dtype)


@derived_from(np)