cascade.nn.ScaleFree.energy

ScaleFree.energy(scaffold)[source]

Scale-free energy function (negative log probability) of a graph scaffold

Parameters:

scaffold (Scaffold) – Graph scaffold

Return type:

Tensor

Returns:

Scale-free energy of shape (n_particles,)