waxmorph.torch.mlp

PyTorch MLP used by the graph network.

Classes

MLP(input_dim, output_dim[, hidden_dim, ...])

MLP with one or more linear layers.