__init__(Z, X, gated_plate=-1, moments=None, **kwargs)¶
Constructor for the gating node.
- Z : Categorical-like node
A variable which chooses the index along the gated plate axis
- X : node
The node whose plate axis is gated
- gated_plate : int (optional)
The index of the plate axis to be gated (by default, -1, that is, the last axis).