def __init__(self, func):
        """
        Creates a torch.nn.Module that applies the function func.
        :param func: lambda function
        """
        super().__init__()
        self.func = func

        Creates a torch.nn.Module that applies the function func.
        :param func: lambda function
        
analyze/convnexts4nd/src/models/baselines/ckconv.py