types.core.ControlPolicy
types.core.ControlPolicy
Control policy/controller π(x).
Maps state to control action: u = π(x).
Parameters
| Name | Type | Description | Default |
|---|---|---|---|
| x | StateVector | Current state (nx,) | required |
Returns
| Name | Type | Description |
|---|---|---|
| ControlVector | Control action (nu,) |
Examples
>>> # LQR policy
>>> def lqr_policy(x: StateVector) -> ControlVector:
... return -K @ x
>>>
>>> # Nonlinear policy
>>> def nonlinear_policy(x: StateVector) -> ControlVector:
... return -np.tanh(K @ x)
>>>
>>> # Neural network policy
>>> def nn_policy(x: StateVector) -> ControlVector:
... return model.forward(torch.tensor(x)).numpy()