CondSeq

class torchelie.nn.CondSeq(*args: torch.nn.modules.module.Module)

class torchelie.nn.CondSeq(arg: OrderedDict[str, Module])

An extension to torch’s Sequential that allows conditioning either as a second forward argument or condition()

condition(z: Any) → None

Conditions all the layers on z

Parameters

z – conditioning

forward(x: Any, z: Optional[Any] = None) → Any

Forward pass

Parameters

• x – input

• z (optional) – conditioning. condition() must be called first if left None

training: bool