Skip to content

Split up MLP blocks vs MLP models

Vicentini Filippo requested to merge github/fork/jwnys/mlp_split into master

Created by: jwnys

As requested by @PhilipVinc , I've put the general MLP (with possible multi-dimensional output) under netket.nn. An additional MLP is available under nk.models.MLP, which has output dimension 1 and automatically squeezes the output dimension. This removes the kwarg squeeze_output we have before, and will allow us to do the same for DeepSets, for example.

Merge request reports