Note that, in input module and communication module, the same types of nodes share parameters, which may speed up the learning process.
in output module, in oder to capture the specific characteristics of each node, they don't share parameters.