keras/src/layers/regularization/spatial_dropout.py/SpatialDropout2D/_get_noise_shape
class SpatialDropout2D:    def _get_noise_shape(self, inputs):
        input_shape = ops.shape(inputs)
        if self.data_format == "channels_first":
            return (input_shape[0], input_shape[1], 1, 1)
        elif self.data_format == "channels_last":
            return (input_shape[0], 1, 1, input_shape[3])