Как я могу удалить или опустить данные с помощью метода карты для объектов tf.data.Dataset?

Я использую тензорный поток 2.3.0

У меня есть генератор данных на Python —

import tensorflow as tf import numpy as np vocab = [1,2,3,4,5] def create_generator(): ‘generates a random number from 0 to len(vocab)-1’ count = 0 while count < 4: x = np.random.randint(0, len(vocab)) yield x count +=1

Я делаю это объектом tf.data.Dataset

gen = tf.data.Dataset.from_generator(create_generator, args=[], output_types=tf.int32, output_shapes = (), )

Теперь я хочу выполнить подвыборку элементов с помощью метода map, чтобы генератор tf никогда не выводил четное число.

def subsample(x): ‘remove item if it is present in an even number [2,4]’ »’ #TODO »’ return x gen = gen.map(subsample)

Как я могу добиться этого с помощью метода map?

Источник: ledsshop.ru