51caa2e2cebe2e6e8c06ffbb918448dd2db011a7,dnc/dnc.py,DNC,_layer_forward,#DNC#Any#Any#Any#Any#,170

Before Change

    (chx, mhx) = hx

    layer_input = input
    hchx = []
    for hlayer in range(self.num_hidden_layers):
      h = self.rnns[layer][hlayer](layer_input, chx[hlayer])
      layer_input = h[0] if self.rnn_type.lower() == "lstm" else h
      hchx.append(h)
    chx = hchx

    // the interface vector
    ξ = layer_input

After Change

    // pass through the controller layer
    input, chx = self.rnns[layer](input.unsqueeze(1), chx)
    input = input.squeeze(1)

    // the interface vector
    ξ = input
    // the output
    output = input

    // pass through memory
    if pass_through_memory:
      if self.share_memory:

Instances