コード例 #1
0
ファイル: ops_test.py プロジェクト: sgcm520/tensorflow2
 def test_sum(self):
     initializer_lt = ops.constant([0, 10], axes=['y'])
     tensor_lt = ops.constant([[1, 2], [3, 4], [5, 6]], axes=['x', 'y'])
     foldl_lt = ops.foldl(core.add, tensor_lt, initializer_lt)
     golden_lt = ops.constant([9, 22], axes=['y'])
     self.assertLabeledTensorsEqual(foldl_lt, golden_lt)
コード例 #2
0
ファイル: ops_test.py プロジェクト: Ajaycs99/tensorflow
 def test_sum(self):
   initializer_lt = ops.constant([0, 10], axes=['y'])
   tensor_lt = ops.constant([[1, 2], [3, 4], [5, 6]], axes=['x', 'y'])
   foldl_lt = ops.foldl(core.add, tensor_lt, initializer_lt)
   golden_lt = ops.constant([9, 22], axes=['y'])
   self.assertLabeledTensorsEqual(foldl_lt, golden_lt)
コード例 #3
0
ファイル: ops_test.py プロジェクト: sgcm520/tensorflow2
 def test_name(self):
     foldl_lt = ops.foldl(core.add, self.original_lt,
                          core.slice_function(self.original_lt, {'x': 0}))
     self.assertIn('lt_foldl', foldl_lt.name)
コード例 #4
0
ファイル: ops_test.py プロジェクト: Ajaycs99/tensorflow
 def test_name(self):
   foldl_lt = ops.foldl(core.add, self.original_lt,
                        core.slice_function(self.original_lt, {'x': 0}))
   self.assertIn('lt_foldl', foldl_lt.name)