Python batch_gather_with_default Exemples

Exemple #1

Fichier : ragged_batch_gather_op_test.py Projet : adit-chandra/tensorflow

 def testRankMismatch(
     self, params, indices, default_value, error):
   params = ragged_factory_ops.constant(params)
   indices = ragged_factory_ops.constant(indices)
   with self.assertRaises(error):
     _ = ragged_batch_gather_with_default_op.batch_gather_with_default(
         params, indices, default_value)

Exemple #2

Fichier : ragged_batch_gather_op_test.py Projet : TheVinhLuong102/tensorflow

 def testRankMismatch(
     self, params, indices, default_value, error):
   params = ragged_factory_ops.constant(params)
   indices = ragged_factory_ops.constant(indices)
   with self.assertRaises(error):
     _ = ragged_batch_gather_with_default_op.batch_gather_with_default(
         params, indices, default_value)

Exemple #3

Fichier : ragged_batch_gather_op_test.py Projet : TheVinhLuong102/tensorflow

 def testRaggedBatchGatherWithDefaultOnTensors(
     self, descr, params, indices, expected, default_value):
   params = constant_op.constant(params)
   indices = constant_op.constant(indices)
   expected = constant_op.constant(expected)
   result = ragged_batch_gather_with_default_op.batch_gather_with_default(
       params, indices, default_value)
   self.assertAllEqual(expected, result)

Exemple #4

Fichier : ragged_batch_gather_op_test.py Projet : adit-chandra/tensorflow

 def testRaggedBatchGatherWithDefaultOnTensors(
     self, descr, params, indices, expected, default_value):
   params = constant_op.constant(params)
   indices = constant_op.constant(indices)
   expected = constant_op.constant(expected)
   result = ragged_batch_gather_with_default_op.batch_gather_with_default(
       params, indices, default_value)
   self.assertAllEqual(expected, result)

Exemple #5

Fichier : ragged_batch_gather_op_test.py Projet : TheVinhLuong102/tensorflow

 def testInvalidDefaultValueRank(
     self, descr, params, indices, default_value, error, ragged_rank=None,
     indices_ragged_rank=None):
   params = ragged_factory_ops.constant(params, ragged_rank=ragged_rank)
   indices = ragged_factory_ops.constant(
       indices, ragged_rank=indices_ragged_rank)
   with self.assertRaises(error):
     _ = ragged_batch_gather_with_default_op.batch_gather_with_default(
         params, indices, default_value)

Exemple #6

Fichier : ragged_batch_gather_op_test.py Projet : adit-chandra/tensorflow

 def testInvalidDefaultValueRank(
     self, descr, params, indices, default_value, error, ragged_rank=None,
     indices_ragged_rank=None):
   params = ragged_factory_ops.constant(params, ragged_rank=ragged_rank)
   indices = ragged_factory_ops.constant(
       indices, ragged_rank=indices_ragged_rank)
   with self.assertRaises(error):
     _ = ragged_batch_gather_with_default_op.batch_gather_with_default(
         params, indices, default_value)

Exemple #7

Fichier : ragged_batch_gather_op_test.py Projet : TheVinhLuong102/tensorflow

 def testRaggedBatchGatherWithDefault(
     self, descr, params, indices, expected, indices_ragged_rank=None,
     expected_ragged_rank=None, ragged_rank=None, default_value='$NONE^'):
   params = ragged_factory_ops.constant(params, ragged_rank=ragged_rank)
   indices = ragged_factory_ops.constant(
       indices, ragged_rank=indices_ragged_rank or ragged_rank)
   expected = ragged_factory_ops.constant(
       expected, ragged_rank=expected_ragged_rank or ragged_rank)
   result = ragged_batch_gather_with_default_op.batch_gather_with_default(
       params, indices, default_value)
   self.assertAllEqual(result, expected)

Exemple #8

Fichier : ragged_batch_gather_op_test.py Projet : adit-chandra/tensorflow

 def testRaggedBatchGatherWithDefault(
     self, descr, params, indices, expected, indices_ragged_rank=None,
     expected_ragged_rank=None, ragged_rank=None, default_value='$NONE^'):
   params = ragged_factory_ops.constant(params, ragged_rank=ragged_rank)
   indices = ragged_factory_ops.constant(
       indices, ragged_rank=indices_ragged_rank or ragged_rank)
   expected = ragged_factory_ops.constant(
       expected, ragged_rank=expected_ragged_rank or ragged_rank)
   result = ragged_batch_gather_with_default_op.batch_gather_with_default(
       params, indices, default_value)
   self.assertRaggedEqual(result, expected)