f7a51531030d64f7a4007deceaf1a2eb19c6e7c1,theanolm/layers/grulayer.py,GRULayer,_create_time_step,#GRULayer#Any#Any#Any#Any#Any#Any#,114

Before Change

        h_out = (1.0 - u) * h_in + u * h_candidate

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

        return h_out

After Change

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

        return h_out

Instances