def get_transform(self, input):
    params = self._get_param_values(use_schedulers=True)
    p = params['p']
    if (self.dims is not None):
        dims = tuple((d for d in self.dims if (self._rand() < p)))
    elif isinstance(p, Dict):
        dims = tuple((k for (k, v) in p.items() if (self._rand() < v)))
    else:
        dims = tuple((d for d in range((- self.ndim), 0) if (self._rand() < p)))
    return (FlipTransform(dims) if dims else NoOpTransform())
def get_transform(self, input):
    params = self._get_param_values(use_schedulers=True)
    p = params['p']
    if (self.dims is not None):
        dims = tuple((d for d in self.dims if (self._rand() < p)))
    elif isinstance(p, Dict):
        dims = tuple((k for (k, v) in p.items() if (self._rand() < v)))
    else:
        dims = tuple((d for d in range((- self.ndim), 0) if (self._rand() < p)))
    return (FlipTransform(dims) if dims else NoOpTransform())