torchcell.models.linear.SimpleLinearModel

class SimpleLinearModel(input_dim: int, output_dim: int, scatter: str = 'add')

Bases: Module

Scatter-aggregate node features per set, then apply a linear layer.

forward(x: Tensor, batch: Tensor) → Tensor

Aggregate node features by batch and project to the output dimension.