Source code for pina.model.multi_feed_forward

"""Module for Multi FeedForward model"""

import torch

from .feed_forward import FeedForward


[docs] class MultiFeedForward(torch.nn.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. :param dict ffn_dict: dictionary of FeedForward networks. """ def __init__(self, ffn_dict): super().__init__() if not isinstance(ffn_dict, dict): raise TypeError for name, constructor_args in ffn_dict.items(): setattr(self, name, FeedForward(**constructor_args))