ott.neural.networks.potentials.PotentialMLP.has_variable

PotentialMLP.has_variable(col, name)

Checks if a variable of given collection and name exists in this Module.

See flax.core.variables for more explanation on variables and collections.

Parameters:

• col (str) – The variable collection name.

• name (str) – The name of the variable.

Return type:

bool

Returns:

True if the variable exists.