Ejemplos de fold en Python

Ejemplo n.º 1

Archivo: control_flow_ops_py_test.py Proyecto: peace195/tensorflow

 def testFold_1(self):
     with self.test_session():
         elems = tf.constant([1, 2, 3, 4, 5, 6], name="data")
         r = control_flow_ops.fold(lambda a, x: tf.mul(tf.add(a, x), 2), elems, [1])
         result = r.eval()
     self.assertTrue(check_op_order(elems.graph))
     self.assertAllEqual(np.array([208]), result)

Ejemplo n.º 2

Archivo: control_flow_ops_py_test.py Proyecto: manniru/intro-to-dl-android

 def testFold_1(self):
   with self.test_session():
     elems = tf.constant([1, 2, 3, 4, 5, 6], name="data")
     r = control_flow_ops.fold(
         lambda a, x: tf.mul(tf.add(a, x), 2), elems, [1])
     result = r.eval()
   self.assertTrue(check_op_order(elems.graph))
   self.assertAllEqual(np.array([208]), result)

Ejemplo n.º 3

Archivo: control_flow_ops_py_test.py Proyecto: hypatiad/tensorflow

  def testFold_2(self):
    with self.test_session():
      elems = tf.constant([1, 2, 3, 4, 5, 6], name="data")
      ten = tf.convert_to_tensor(10)

      def compute(a, x):
        r = tf.mul(x, ten)
        return tf.add(a, r)

      r = control_flow_ops.fold(compute, elems, [1])
      result = r.eval()
    self.assertTrue(check_op_order(elems.graph))
    self.assertAllEqual([201], result)

Ejemplo n.º 4

Archivo: control_flow_ops_py_test.py Proyecto: manniru/intro-to-dl-android

  def testFold_2(self):
    with self.test_session():
      elems = tf.constant([1, 2, 3, 4, 5, 6], name="data")
      ten = tf.convert_to_tensor(10)

      def compute(a, x):
        r = tf.mul(x, ten)
        return tf.add(a, r)

      r = control_flow_ops.fold(compute, elems, [1])
      result = r.eval()
    self.assertTrue(check_op_order(elems.graph))
    self.assertAllEqual([201], result)