def METHOD_NAME(self, input_shape):
    assert len(input_shape) >= 2
    layer_kwargs = dict(
        kernel_initializer=self.kernel_initializer,
        bias_initializer=self.bias_initializer,
        kernel_regularizer=self.kernel_regularizer,
        bias_regularizer=self.bias_regularizer,
        kernel_constraint=self.kernel_constraint,
        bias_constraint=self.bias_constraint,
        dtype=self.dtype,
    )
    mlp_layers = []
    for channels in self.mlp_hidden:
        mlp_layers.extend(
            [
                Dropout(self.dropout_rate),
                Dense(channels, self.mlp_activation, **layer_kwargs),
            ]
        )
    mlp_layers.append(Dense(self.channels, "linear", **layer_kwargs))
    self.mlp = Sequential(mlp_layers)
    self.built = True