test_decoder_rnn.py 文件源码

def test_dropout_WITH_PROB_ZERO(self):
        rnn = DecoderRNN(self.vocab_size, 50, 16, 0, 1, dropout_p=0)
        for param in rnn.parameters():
            param.data.uniform_(-1, 1)
        output1, _, _ = rnn()
        output2, _, _ = rnn()
        for prob1, prob2 in zip(output1, output2):
            self.assertTrue(torch.equal(prob1.data, prob2.data))