Gate.__init__(Z, X, gated_plate=-1, moments=None, **kwargs)[source]

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).