dgs.models.alpha.alpha

Base class for modules that predict alpha values given a State.

Module Classes

BaseAlphaModule(*args, **kwargs)

Given a state as input, compute and return the weight of the alpha gate.