Support tensors as dropout rates again, by removing the min(max(..)..) clipping

that throws TypeError for tensors.

PiperOrigin-RevId: 157000504

Author: tensorflower-gardener

tensorflow/python/layers/core.py

@@ -244,7 +244,7 @@ def __init__(self, rate=0.5,
                name=None,
                **kwargs):
     super(Dropout, self).__init__(name=name, **kwargs)
-    self.rate = min(1., max(0., rate))
+    self.rate = rate
     self.noise_shape = noise_shape
     self.seed = seed
 

tensorflow/python/layers/core_test.py

@@ -337,6 +337,18 @@ def testFunctionalDropout(self):
       np_output = sess.run(dropped, feed_dict={training: False})
       self.assertAllClose(np.ones((5, 5)), np_output)
 
+  def testDynamicRate(self):
+    with self.test_session() as sess:
+      rate = array_ops.placeholder(dtype='float32', name='rate')
+      dp = core_layers.Dropout(rate, name='dropout')
+      inputs = array_ops.ones((5, 5))
+      dropped = dp.apply(inputs, training=True)
+      sess.run(variables.global_variables_initializer())
+      np_output = sess.run(dropped, feed_dict={rate: 0.5})
+      self.assertAlmostEqual(0., np_output.min())
+      np_output = sess.run(dropped, feed_dict={rate: 0.0})
+      self.assertAllClose(np.ones((5, 5)), np_output)
+
 
 if __name__ == '__main__':
   test.main()