f7a51531030d64f7a4007deceaf1a2eb19c6e7c1,theanolm/layers/lstmlayer.py,LSTMLayer,_create_time_step,#LSTMLayer#Any#Any#Any#Any#Any#Any#Any#,119

Before Change

        h_out = o * tensor.tanh(C_out)

        // Apply the mask.
        C_out = mask[:, None] * C_out + (1.0 - mask)[:, None] * C_in
        h_out = mask[:, None] * h_out + (1.0 - mask)[:, None] * h_in

        return C_out, h_out

After Change

// XXX        C_out = mask[:,None] * C_out + (1.0 - mask)[:,None] * C_in
// XXX        h_out = mask[:,None] * h_out + (1.0 - mask)[:,None] * h_in
        C_out = tensor.switch(mask[:,None], C_out, C_in)
        h_out = tensor.switch(mask[:,None], h_out, h_in)

        return C_out, h_out

Instances