Method forward

climart/models/modules/mlp.py:94–98

(self, X: Tensor)

            print('MLP block with residual!')

    def forward(self, X: Tensor) -> Tensor:
        X_out = self.layer(X)
        if self.residual:
            X_out += X
        return X_out