torchcell.models.hetero_cell_bipartite_dango.PreProcessor

class PreProcessor(in_channels: int, hidden_channels: int, num_layers: int = 2, dropout: float = 0.1, norm: str = 'layer', activation: str = 'relu')

Bases: Module

MLP that projects raw input features to the hidden dimension.

forward(x: Tensor) → Tensor

Apply the preprocessing MLP to the input features.