Language Language English 简体中文

unilab.base.reset_selectors.resolve_reset_mask

unilab.base.reset_selectors.resolve_reset_mask(selector, *, num_envs, device, owner)

Resolve a public reset selector to the internal fixed-width bool mask.

The selector is normalized to env_ids and scattered into a freshly allocated mask.

Parameters:

• selector (Tensor | slice | None)

• num_envs (int)

• device (device)

• owner (str)

Return type:

Tensor