def diagonal(a, offset=0, axis1=0, axis2=1):
    # TODO(okuta): check type
    return a.diagonal(offset, axis1, axis2)


# TODO(okuta): Implement select
Returns specified diagonals.
dataset/ETHPy150Open pfnet/chainer/cupy/indexing/indexing.py/diagonal