mr2.operators.models.BMC.ResetBlock

class mr2.operators.models.BMC.ResetBlock[source]

Bases: BMCBlock

Reset state to equilibrium or to a provided state.

__init__(state: Tensor | None = None) → None[source]

Initialize the block.

Parameters:

state (Tensor | None, default: None) – State tensor. Shape (..., pools, 3).

property duration: Tensor[source]

Duration of the block.

__call__(parameters: Parameters, state: Tensor | None = None, *, zero_matrix: Tensor | None = None, zero_c: Tensor | None = None) → tuple[Tensor, tuple[Tensor, ...]][source]

Apply the block.

forward(parameters: Parameters, state: Tensor) → tuple[Tensor, tuple[Tensor, ...]][source]

Apply the block.

Parameters:

• parameters (Parameters) – Simulation parameters.

• state (Tensor) – State tensor. Shape (..., isochromats, pools, 3).

Returns:

state – State tensor. Shape (..., isochromats, pools, 3).