ccbc21938d895de333e83a665ec6147adabf35c2,sru/sru_functional.py,SRUCell,forward,#SRUCell#,268

Before Change

            mask_h = self.get_dropout_mask_((batch, n_out*bidir), self.dropout)
            return SRU_Compute(u, input, self.weight_c, self.bias, c0, mask_h)
        else:
            return SRU_Compute(u, input, self.weight_c, self.bias, c0)

    def get_dropout_mask_(self, size, p):
        
        Composes the dropout mask for the `SRUCell`.

After Change

            mask_h = self.get_dropout_mask_((batch, n_out*bidir), self.dropout)
            h, c = SRU_Compute(u, input, self.weight_c, self.bias, c0, mask_h)
        else:
            h, c = SRU_Compute(u, input, self.weight_c, self.bias, c0)

        if return_proj:
            return h, c, x_projected.view(-1, batch, self.n_proj)
        else:

Instances