(transform_num: int)
return policy_id, probs, signs
Get parameters for autoaugment transformation

        Returns:
            params required by the autoaugment transformation
Get parameters for autoaugment transformation
Get parameters for autoaugment transformation
def get_params(transform_num: int) -> Tuple[int, Tensor, Tensor]:
        """Get parameters for autoaugment transformation

        Returns:
            params required by the autoaugment transformation
        """
        policy_id = torch.randint(transform_num, (1, )).item()
        probs = torch.rand((2, ))
        signs = torch.randint(2, (2, ))

        return policy_id, probs, signs
def get_params ( transform_num : int ) -> Tuple [ int , Tensor , Tensor ] : policy_id = torch . randint ( transform_num , ( 1 , ) ) . item ( ) probs = torch . rand ( ( 2 , ) ) signs = torch . randint ( 2 , ( 2 , ) ) return policy_id , probs , signs