Language

unilab.base.reset_selectors.resolve_reset_env_ids¶

unilab.base.reset_selectors.resolve_reset_env_ids(selector, *, num_envs, device, owner)[source]¶

Normalize a public reset selector to an int64 row-index tensor.

None selects every row; a slice selects arange(num_envs)[slice]; an integer tensor is validated and moved onto device. Fixed-width bool masks are rejected: the v1.4 mask form was removed in v1.5.

Parameters:

selector (Tensor | slice | None)
num_envs (int)
device (device)
owner (str)

Return type:

Tensor