Neural network 如何在Pytork中生成非分层前馈NN？

我想建立一个网络，它可能有很多复杂的跳过连接，但仍然是前馈的（没有重复连接）。每个节点仍然是一个“经典”节点，因为它对每个连接都有一个权重，然后将每个权重乘以相应的输入，求和，并将其放入一个非线性函数

例如，它可能如下所示：

但是，我不认为我可以轻松地使用

nn.Linear

模块来计算输出，因为（例如，在图片中），红色节点从绿色节点和粉色节点获取输入，但绿色节点也需要粉色节点的输出来计算

我可以想出一些方法来做到这一点，但它们真的很难看，pytorch通常有一种聪明的做事方式。有什么好办法吗？

你可能想看看。你可能想看看。