bd570b58cefa57062f8c14e06b83800c52f4925a,sru/sru_functional.py,SRU,forward,#SRU#Any#Any#Any#Any#,545

Before Change

        if return_hidden:
            print("lstc[0].size()", lstc[0].size())
            lstc_stack = torch.stack(lstc)
            if self.nn_rnn_compatible_return:
                print("lstc_stack.size()", lstc_stack.size())
            return prevx, lstc_stack
        else:
            return prevx

After Change

        if return_hidden:
            lstc_stack = torch.stack(lstc)
            if self.nn_rnn_compatible_return:
                batch_size = input.size(1)
                lstc_stack = lstc_stack.view(self.num_layers, batch_size, self.num_directions, self.n_out)
                lstc_stack = lstc_stack.transpose(1, 2)
                lstc_stack = lstc_stack.contiguous().view(self.num_layers * self.num_directions, batch_size, self.n_out)
            return prevx, lstc_stack
        else:
            return prevx

Instances