torchcell.models.dcell.DCellSubsystem

class DCellSubsystem(input_dim: int, output_dim: int)

Bases: Module

Individual subsystem module as described in DCell paper.

forward(x: Tensor) → Tensor

Apply linear, batch norm, and tanh to the input.