22 lines
539 B
Python
22 lines
539 B
Python
"""Module for Multi FeedForward model"""
|
|
import torch
|
|
|
|
from .feed_forward import FeedForward
|
|
|
|
|
|
class MultiFeedForward(torch.nn.Module):
|
|
"""
|
|
:param dict dff_dict: dictionary of FeedForward networks.
|
|
"""
|
|
def __init__(self, dff_dict):
|
|
'''
|
|
dff_dict: dict of FeedForward objects
|
|
'''
|
|
super().__init__()
|
|
|
|
if not isinstance(dff_dict, dict):
|
|
raise TypeError
|
|
|
|
for name, constructor_args in dff_dict.items():
|
|
setattr(self, name, FeedForward(**constructor_args))
|