From 5a08a09754ac08a6870217432542a899f58a50b5 Mon Sep 17 00:00:00 2001 From: Graham Neubig Date: Wed, 30 Nov 2016 16:04:37 -0500 Subject: [PATCH] Removed extra multiply --- dynet/lstm.cc | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/dynet/lstm.cc b/dynet/lstm.cc index 55aef93c0..ec0f51919 100644 --- a/dynet/lstm.cc +++ b/dynet/lstm.cc @@ -93,17 +93,17 @@ void LSTMBuilder::start_new_sequence_impl(const vector& hinit) { float retention_rate = 1.f - dropout_rate; float scale = 1.f / retention_rate; // in - masks_i.push_back(random_bernoulli(*_cg,{ idim}, retention_rate) * scale); - masks_i.push_back(random_bernoulli(*_cg,{ idim}, retention_rate) * scale); - masks_i.push_back(random_bernoulli(*_cg,{ idim}, retention_rate) * scale); + masks_i.push_back(random_bernoulli(*_cg,{ idim}, retention_rate, scale)); + masks_i.push_back(random_bernoulli(*_cg,{ idim}, retention_rate, scale)); + masks_i.push_back(random_bernoulli(*_cg,{ idim}, retention_rate, scale)); // h - masks_i.push_back(random_bernoulli(*_cg,{ hidden_dim}, retention_rate) * scale); - masks_i.push_back(random_bernoulli(*_cg,{ hidden_dim}, retention_rate) * scale); - masks_i.push_back(random_bernoulli(*_cg,{ hidden_dim}, retention_rate) * scale); + masks_i.push_back(random_bernoulli(*_cg,{ hidden_dim}, retention_rate, scale)); + masks_i.push_back(random_bernoulli(*_cg,{ hidden_dim}, retention_rate, scale)); + masks_i.push_back(random_bernoulli(*_cg,{ hidden_dim}, retention_rate, scale)); // c - masks_i.push_back(random_bernoulli(*_cg,{ hidden_dim}, retention_rate) * scale); - masks_i.push_back(random_bernoulli(*_cg,{ hidden_dim}, retention_rate) * scale); - masks_i.push_back(random_bernoulli(*_cg,{ hidden_dim}, retention_rate) * scale); + masks_i.push_back(random_bernoulli(*_cg,{ hidden_dim}, retention_rate, scale)); + masks_i.push_back(random_bernoulli(*_cg,{ hidden_dim}, retention_rate, scale)); + masks_i.push_back(random_bernoulli(*_cg,{ hidden_dim}, retention_rate, scale)); masks.push_back(masks_i); } }