cascade.nn.TrExp.energy

TrExp.energy(scaffold)

Tr-Exp acyclic energy function (negative log probability) of a graph scaffold

Parameters:

scaffold (Scaffold) – Graph scaffold

Return type:

Tensor

Returns:

Tr-Exp acyclic energy of shape (n_particles,)