d3rlpy.augmentation.image.RandomShift¶
-
class
d3rlpy.augmentation.image.
RandomShift
(shift_size=4)[source]¶ Random shift augmentation.
References
Parameters: shift_size (int) – size to shift image. Methods
-
get_params
(deep=False)[source]¶ Returns augmentation parameters.
Parameters: deep (bool) – flag to deeply copy objects. Returns: augmentation parameters. Return type: dict
-
transform
(x)[source]¶ Returns shifted images.
Parameters: x (torch.Tensor) – observation tensor. Returns: processed observation tensor. Return type: torch.Tensor
Attributes
-
TYPE
= 'random_shift'¶
-