Hi there,
I am new to PyTorch, but I want to create a verysimple linear layer with custom connections.
As an example, I want to create the following connectivity:

My code does not work:
class Net(nn.Module):
  def __init__(self):
    super(Net, self).__init__()
    self.l1 = nn.Linear(n_feats, 3, bias=False)
    self.t = Variable(torch.randn(N_BATCH, 2))
    self.neurons = [nn.Linear(2, 1), nn.Linear(2, 1)]

  def forward(self, x):
    x = F.relu(self.l1(x))
    self.t[:, 0] = self.neurons[0](x[:, :2])
    self.t[:, 1] = self.neurons[1](x[:, 1:])
    return t

Besides being ugly, the I expect to have (n_feats x 3 + 2 * 2 + 2) parameters. My model displays only (n_feats x 3) parameters.
What is the correct way to do this?