keras/src/legacy/layers.py/RandomHeight/compute_output_shape
class RandomHeight:    def compute_output_shape(self, input_shape):
        input_shape = list(input_shape)
        input_shape[-3] = None
        return tuple(input_shape)