Python _check_static_batch_beam_maybe Exemples

Exemple #1

 def test_array_shape_static_checks(self):
     self.assertTrue(
         beam_search_decoder._check_static_batch_beam_maybe(
             tf.TensorShape([None, None, None]), 3, 5
         )
     )
     self.assertTrue(
         beam_search_decoder._check_static_batch_beam_maybe(
             tf.TensorShape([15, None, None]), 3, 5
         )
     )
     self.assertFalse(
         beam_search_decoder._check_static_batch_beam_maybe(
             tf.TensorShape([16, None, None]), 3, 5
         )
     )
     self.assertTrue(
         beam_search_decoder._check_static_batch_beam_maybe(
             tf.TensorShape([3, 5, None]), 3, 5
         )
     )
     self.assertFalse(
         beam_search_decoder._check_static_batch_beam_maybe(
             tf.TensorShape([3, 6, None]), 3, 5
         )
     )
     self.assertFalse(
         beam_search_decoder._check_static_batch_beam_maybe(
             tf.TensorShape([5, 3, None]), 3, 5
         )
     )

Exemple #2

Fichier : beam_search_decoder_test.py Projet : xvr-hlt/addons

def test_array_shape_static_checks():
    assert (beam_search_decoder._check_static_batch_beam_maybe(
        tf.TensorShape([None, None, None]), 3, 5) is True)

    assert (beam_search_decoder._check_static_batch_beam_maybe(
        tf.TensorShape([15, None, None]), 3, 5) is True)
    assert (beam_search_decoder._check_static_batch_beam_maybe(
        tf.TensorShape([16, None, None]), 3, 5) is False)
    assert (beam_search_decoder._check_static_batch_beam_maybe(
        tf.TensorShape([3, 5, None]), 3, 5) is True)
    assert (beam_search_decoder._check_static_batch_beam_maybe(
        tf.TensorShape([3, 6, None]), 3, 5) is False)
    assert (beam_search_decoder._check_static_batch_beam_maybe(
        tf.TensorShape([5, 3, None]), 3, 5) is False)