dgs.utils.state.State.extract

State.extract(idx: int) → State

Extract the i-th State from a batch B of states.

Parameters:

idx – The index of the State to retrieve. It is expected that \(-B \lte idx \lt B\).

Returns:

The extracted State.