cellflow.networks.ResNetBlock.__call__¶
- ResNetBlock.__call__(x, cond, *, training=True)[source]¶
Forward pass of the residual layer.
- Parameters:
x (
Array) – Input features of shape (batch, input_dim).cond (
Array) – Conditioning features of shape (batch, cond_dim).training (
bool) – Whether the model is in training mode.
- Return type:
Array- Returns:
: Output features of shape (batch, input_dim).