cellflow.networks.ConditionalVelocityField.copy

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

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

Return type:

TypeVar(M, bound= Module)

Parameters:
  • self (M)

  • parent (Scope | Module | _Sentinel | None)

  • name (str | None)

Args:
parent: The parent of the copy. By default the current module is taken

as parent if not explicitly specified.

name: A new name for the copied Module, by default a new automatic name

will be given.

**updates: Attribute updates.

Returns:

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