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?