keras/src/ops/numpy.py/Argmin/compute_output_spec
class Argmin:    def compute_output_spec(self, x):
        if self.keepdims:
            return KerasTensor(x.shape, dtype="int32")
        if self.axis is None:
            return KerasTensor([], dtype="int32")
        return KerasTensor(
            reduce_shape(x.shape, axis=[self.axis]), dtype="int32"
        )