如何在pytorch中制作非分层前馈NN？

Question

我想建立一个网络，该网络可能具有许多复杂的跳过连接，但仍会前馈（无循环连接）。每个节点仍将是每个连接的权重，然后将每个权重乘以相应的输入，对其求和，然后通过非线性函数将其视为“经典”节点。

例如，它可能看起来像这样：

但是，我不认为我可以轻易使用nn.Linear模块之类的东西来计算输出，因为（例如，在图片中）红色节点从绿色节点和< / em>粉色节点，但是绿色节点也需要粉色节点的输出进行计算。

我可以想到一些方法来做到这一点，但是它们确实很丑陋，而pytorch通常有一个聪明的做事方法。有什么好方法吗？