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