def get_parameter_projection(self, in_features: int) -> nn.Module:
        r"""
        Return the parameter projection layer that maps the input to the appropriate parameters of the distribution.
        """
        return ParameterProjection(
            in_features=in_features,
            args_dim=self.args_dim,
            domain_map=LambdaLayer(self.domain_map),
        )

        Return the parameter projection layer that maps the input to the appropriate parameters of the distribution.
        
dpkd/transformers/src/transformers/time_series_utils.py