keras/src/layers/regularization/spatial_dropout.py/SpatialDropout1D/_get_noise_shape
class SpatialDropout1D:    def _get_noise_shape(self, inputs):
        input_shape = ops.shape(inputs)
        return (input_shape[0], 1, input_shape[2])