MultiFeedForward

class MultiFeedForward(ffn_dict)

Bases: Module

The PINA implementation of MultiFeedForward network.

This model allows to create a network with multiple FeedForward combined together. The user has to define the forward method choosing how to combine the different FeedForward networks.

Parameters:

ffn_dict (dict) – dictionary of FeedForward networks.

Initialize internal Module state, shared by both nn.Module and ScriptModule.