예제 #1
0
 def __init__(
     self,
     kernel_size: Tuple[int, int],
     sigma: Tuple[float, float],
     border_type: str = "reflect",
     same_on_batch: bool = False,
     p: float = 0.5,
     keepdim: bool = False,
     return_transform: Optional[bool] = None,
 ) -> None:
     super().__init__(p=p,
                      return_transform=return_transform,
                      same_on_batch=same_on_batch,
                      p_batch=1.0,
                      keepdim=keepdim)
     self.flags = dict(kernel_size=kernel_size,
                       sigma=sigma,
                       border_type=BorderType.get(border_type))