keras/src/ops/numpy.py/Select/compute_output_spec
class Select:    def compute_output_spec(self, condlist, choicelist, default=0):
        first_element = choicelist[0]
        return KerasTensor(first_element.shape, dtype=first_element.dtype)