unilab.base.augmentation.SymmetryAugmentation

class unilab.base.augmentation.SymmetryAugmentation[source]

Bases: Protocol

Runtime symmetry augmentation contract owned by env/backend adapters.

Methods

__init__(*args, **kwargs)

augment_obs_and_actions(obs, actions, *[, ...])

mirror_obs(obs, *[, obs_group])

Attributes

batch_multiplier

batch_multiplier: int
augment_obs_and_actions(obs, actions, *, obs_group='obs')[source]
Parameters:
Return type:

tuple[Tensor, Tensor]

mirror_obs(obs, *, obs_group='obs')[source]
Parameters:
Return type:

Tensor

__init__(*args, **kwargs)