unilab.envs.mdp.events.resolve_env_ids

unilab.envs.mdp.events.resolve_env_ids(env, env_ids)[source]

Return int64 env rows on the env device; no host detour for device envs.

Parameters:
Return type:

Tensor