Topology flow. Current implementation has an Error.

[yoookoo]

In your implementation line 71~73 in model.py , nodes are divided into input and first-order ones. This may lead that second(or higher)-order(from input node) ones are ignored. Your implementation only consider the situation that the max distance between input and output node is 2. Am I right?

[yoookoo]

Maybe we should calculate through Topology structure not through (input/mid/output) division.

[yoookoo]

Coincidentally, your implementation happens to be a dropout effect.