torchrl.trainers.algorithms.configs.transforms.ToTensorImageConfig

class torchrl.trainers.algorithms.configs.transforms.ToTensorImageConfig(from_int: bool | None = None, unsqueeze: bool = False, dtype: str | None = None, in_keys: list[str] | None = None, out_keys: list[str] | None = None, shape_tolerant: bool = False, _target_: str = 'torchrl.envs.transforms.transforms.ToTensorImage')

Configuration for ToTensorImage transform.