ott.solvers.nn.models.ModelBase.clone#

ModelBase.clone(*, parent=None, **updates)#

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

Parameters:
  • parent (Union[Scope, Module, None]) – The parent of the clone. The clone will have no parent if no explicit parent is specified.

  • **updates – Attribute updates.

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

Return type:

TypeVar(M, bound= Module)

Returns:

A clone of the this Module with the updated attributes and parent.