Python dataset_as_numpy Beispiele

Beispiel #1

 def testDatasetAsNumpy(self):
   arrs = extensions.dataset_as_numpy(
       [tf.constant([1, 2]), tf.constant([3, 4])])
   for a in arrs:
     self.assertIsInstance(a, arrays.ndarray)
   with self.assertRaisesWithPredicateMatch(
       ValueError,
       r"dataset_as_numpy must be run in eager mode outside tf.function"):
     @tf.function
     def f():
       return extensions.dataset_as_numpy([tf.constant([1, 2])])
     f()

Beispiel #2

Datei: extensions_test.py Projekt: yliu45/trax

 def f():
   return extensions.dataset_as_numpy([tf.constant([1, 2])])