keras/src/ops/numpy.py/Diagonal/call
class Diagonal:    def call(self, x):
        return backend.numpy.diagonal(
            x,
            offset=self.offset,
            axis1=self.axis1,
            axis2=self.axis2,
        )