Removed extra multiply

clab · Nov 30, 2016 · 5a08a09 · 5a08a09
1 parent 8b441cc
commit 5a08a09
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/dynet/lstm.cc b/dynet/lstm.cc
@@ -93,17 +93,17 @@ void LSTMBuilder::start_new_sequence_impl(const vector<Expression>& 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);
     }
   }