import torch
import torch.nn as nn
import torch.nn.functional as F
class (nn.module):
    def __init__(self, args):
        self.arg = args;
    def forward(self, input_):
        '''define the model'''
    def initNNcell(self):
        '''if necessary'''