cellflow.networks.MLPBlock.__call__

MLPBlock.__call__(x, training=True)

Forward pass.

Parameters:

• x (Array) – Input tensor of shape (batch_size, input_dim).

• training (bool) – Whether the model is in training mode.

Return type:

Array

Returns:

: Output tensor of shape (batch_size, output_dim).