ott.neural.networks.potentials.BasePotential.copy

ott.neural.networks.potentials.BasePotential.copy#

BasePotential.copy(*, parent=<flax.linen.module._Sentinel object>, name=None, **updates)#

Creates a copy of this Module, with optionally updated arguments.

Parameters:

  • parent (Union[Scope, Module, _Sentinel, None]) – The parent of the copy. By default the current module is taken as parent if not explicitly specified.

  • name (str | None) – A new name for the copied Module, by default a new automatic name will be given.

  • **updates – Attribute updates.

  • self (TypeVar(M, bound= Module))

Return type:

TypeVar(M, bound= Module)

Returns:

A copy of the this Module with the updated name, parent, and attributes.