Method _apply_transform

advanced_source/pendulum.py:700–701

(self, obs: torch.Tensor)

class CosTransform(Transform):
    def _apply_transform(self, obs: torch.Tensor) -> None:
        return obs.cos()

    # The transform must also modify the data at reset time
    def _reset(