コード例 #1
0
ファイル: math_ops_test.py プロジェクト: yuikns/tensorflow
 def testFloorDivGrad(self):
     with self.test_session():
         a = variables.Variable(2.0)
         b = variables.Variable(4.0)
         with self.test_session() as sess:
             sess.run(variables.initialize_all_variables())
             c_grad = gradients.gradients(math_ops.div_deprecated(a, b), [a, b])
             self.assertAllEqual([x.eval() for x in c_grad], [0.25, -0.125])
             c_grad = gradients.gradients(math_ops.div(a, b), [a, b])
             self.assertAllEqual([x.eval() for x in c_grad], [0.25, -0.125])
             c_grad = gradients.gradients(math_ops.floordiv(a, b), [a, b])
             self.assertAllEqual([None if x is None else x.eval() for x in c_grad], [None, None])
コード例 #2
0
ファイル: math_ops_test.py プロジェクト: yuikns/tensorflow
 def testComplexDiv(self):
     foo = array_ops.constant([1.0 + 3.0j])
     with self.test_session():
         _ = math_ops.div_deprecated(foo, 1.0).eval()
         _ = math_ops.div(foo, 2.0).eval()