bc714964ae754c25af66ded19ae2926768362584,dnc/dnc.py,DNC,_layer_forward,#DNC#Any#Any#Any#,153

Before Change

      // print("input[time]", input[time].size(), self.layer0_input_size, self.layern_input_size)
      chx = self.rnns[layer](input[time], chx)
      // the interface vector
      ξ = chx[0] if self.rnn_type.lower() == "lstm" else chx
      // the output
      out = self.output_weights(chx[0]) if self.rnn_type.lower() == "lstm" else self.output_weights(chx)

      // pass through memory

After Change

      layer_input = input[time]
      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

Instances