nami.solvers.heun

Classes

Heun([steps])

class nami.solvers.heun.Heun(steps=32)

Bases: object

Parameters:

steps (int)

integrate(f, x0, *, t0, t1, steps=None)

Return type:

Tensor

Parameters:

• x0 (Tensor)

• t0 (float)

• t1 (float)

• steps (int | None)

integrate_augmented(f_aug, x0, logp0, *, t0, t1, steps=None)

Return type:

tuple[Tensor, Tensor]

Parameters:

• x0 (Tensor)

• logp0 (Tensor)

• t0 (float)

• t1 (float)

• steps (int | None)

is_sde = False

requires_steps = True

supports_rsample = True