Quiero usartf.cond (pred, fn1, fn2, nombre = Ninguno) para ramificación condicional. Digamos que tengo dos tensores:x, y. Cada tensor es un lote de 0/1 y quiero usar esta compresión de tensoresx < y como la fuente detf.cond pred argumento:

pred: un escalar que determina si se devuelve el resultado de fn1 o fn2.

Pero si estoy trabajando con lotes, entonces parece que necesito iterar sobre el tensor de origen dentro del gráfico y hacer cortes para cada elemento del lote y aplicar tf.cond para cada elemento. Parece sospechoso para mí. ¿Por qué tf.cond no acepta lotes y solo escalares? ¿Me puede aconsejar cuál es la forma correcta de usarlo con el lote?