keras/src/ops/numpy.py/Identity/compute_output_spec
class Identity:    def compute_output_spec(self, n, dtype=None):
        dtype = dtype or backend.floatx()
        return KerasTensor([n, n], dtype=dtype)