GDSolver

class GDSolver(model: Costable, n_steps: int, batch_size: int | None = None, action_noise: float = 0.0, num_samples: int = 1, device='cpu', seed: int = 1234)[source]

Bases: Module

Gradient Descent Solver.

property action_dim: int
configure(*, action_space, n_envs: int, config) → None[source]
property horizon: int
init_action(actions=None)[source]

Initialize the action tensor for the solver.

set self.init - initial action sequences (n_envs, horizon, action_dim)

property n_envs: int
solve(info_dict, init_action=None) → dict[source]

Solve the planning optimization problem using gradient descent with batch processing.