import torch
import torch.nn as nn

class NsoltSynthesis2dNetwork(nn.Module):

    def __init__(self,
        number_of_channels=[],
        decimation_factor=[]):
        super(NsoltSynthesis2dNetwork, self).__init__()
        self.number_of_channels = number_of_channels
        self.decimation_factor = decimation_factor

    def forward(self,x):
        return x

import torch
import torch.nn as nn
from nsoltBlockIdct2dLayer import NsoltBlockIdct2dLayer
from nsoltFinalRotation2dLayer import NsoltFinalRotation2dLayer 


class NsoltSynthesis2dNetwork(nn.Module):

    def __init__(self,
        number_of_channels=[],
        decimation_factor=[]):
        super(NsoltSynthesis2dNetwork, self).__init__()
        self.number_of_channels = number_of_channels
        self.decimation_factor = decimation_factor
        
        # Instantiation of layers
        self.layerV0 = NsoltFinalRotation2dLayer(
            number_of_channels=number_of_channels,
            decimation_factor=decimation_factor,
            name='V0'
        )
        self.layerE0 = NsoltBlockIdct2dLayer(
            decimation_factor=decimation_factor,
            name='E0'
        )

    def forward(self,x):
        u = self.layerV0.forward(x)
        y = self.layerE0.forward(u) 
        return y

Update nsoltSynthesis2dNetwork.py