Python batch_gather Examples

Example #1

File: ragged_batch_gather_op_test.py Project: JonathanRaiman/tensorflow

 def testRaggedBatchGatherStaticError(self,
                                      params,
                                      indices,
                                      message,
                                      error=ValueError):
   with self.assertRaisesRegexp(error, message):
     ragged.batch_gather(params, indices)

Example #2

File: ragged_batch_gather_op_test.py Project: d813s909q/tensortflow

 def testRaggedBatchGatherStaticError(self,
                                      params,
                                      indices,
                                      message=None,
                                      error=ValueError):
     with self.assertRaisesRegexp(error, message):
         ragged.batch_gather(params, indices)

Example #3

File: ragged_batch_gather_op_test.py Project: JonathanRaiman/tensorflow

  def testRaggedBatchGatherUnknownRankError(self):
    params = [['a', 'b'], ['c', 'd']]
    indices = array_ops.placeholder(dtypes.int32, shape=None)
    ragged_indices = ragged.from_row_splits(indices, [0, 2, 4])

    with self.assertRaisesRegexp(
        ValueError, 'batch_gather does not allow indices with unknown shape.'):
      ragged.batch_gather(params, indices)

    with self.assertRaisesRegexp(
        ValueError, 'batch_gather does not allow indices with unknown shape.'):
      ragged.batch_gather(params, ragged_indices)

Example #4

  def testRaggedBatchGatherUnknownRankError(self):
    params = [['a', 'b'], ['c', 'd']]
    indices = array_ops.placeholder(dtypes.int32, shape=None)
    ragged_indices = ragged.from_row_splits(indices, [0, 2, 4])

    with self.assertRaisesRegexp(
        ValueError, 'batch_gather does not allow indices with unknown shape.'):
      ragged.batch_gather(params, indices)

    with self.assertRaisesRegexp(
        ValueError, 'batch_gather does not allow indices with unknown shape.'):
      ragged.batch_gather(params, ragged_indices)

Example #5

File: ragged_batch_gather_op_test.py Project: JonathanRaiman/tensorflow

 def testRaggedBatchGather(self, descr, params, indices, expected):
   result = ragged.batch_gather(params, indices)
   self.assertEqual(
       getattr(result, 'ragged_rank', 0), getattr(expected, 'ragged_rank', 0))
   with self.test_session():
     if hasattr(expected, 'tolist'):
       expected = expected.tolist()
     self.assertEqual(result.eval().tolist(), expected)

Example #6

File: ragged_batch_gather_op_test.py Project: qwerzou1/shibie

 def testRaggedBatchGather(self, descr, params, indices, expected):
     result = ragged.batch_gather(params, indices)
     self.assertEqual(getattr(result, 'ragged_rank', 0),
                      getattr(expected, 'ragged_rank', 0))
     with self.test_session():
         if hasattr(expected, 'tolist'):
             expected = expected.tolist()
         self.assertEqual(result.eval().tolist(), expected)

Example #7

File: ragged_batch_gather_op_test.py Project: d813s909q/tensortflow

 def testRaggedBatchGather(self, descr, params, indices, expected):
     result = ragged.batch_gather(params, indices)
     self.assertRaggedEqual(result, expected)