Allow for feature-specific distribution

[pat-alt]

Currently, the feature distribution $\mathcal{D}_x$ can only be determined for the whole input matrix. It might be worth adding support for feature-by-feature distributions.