Como usar o tf.cond para processamento em lote

Eu quero usartf.cond (pred, fn1, fn2, nome = Nenhum) para ramificação condicional. Digamos que eu tenha dois tensores:x, y. Cada tensor é um lote de 0/1 e eu quero usar essa compressão de tensoresx < y como fonte detf.cond pred argumento:

pred: Um escalar que determina se deve retornar o resultado de fn1 ou fn2.

Mas se estou trabalhando com lotes, parece que eu preciso iterar sobre o tensor de origem dentro do gráfico e fazer fatias para cada item do lote e aplicar tf.cond para cada item. Parece desconfiado quanto a mim. Por que o tf.cond não aceita lote e apenas escalar? Você pode aconselhar qual é a maneira correta de usá-lo em lote?

questionAnswers(1)

yourAnswerToTheQuestion