def get_transform(self, input):
    params = self._get_param_values(use_schedulers=True)
    if (self._rand() >= params['p']):
        return NoOpTransform()
    k = self.ks[torch.randperm(len(self.ks))[0].item()]
    return Rot90Transform(k=k, dims=((- 1), (- 2)))
def get_transform(self, input):
    params = self._get_param_values(use_schedulers=True)
    if (self._rand() >= params['p']):
        return NoOpTransform()
    k = self.ks[torch.randperm(len(self.ks))[0].item()]
    return Rot90Transform(k=k, dims=((- 1), (- 2)))