Да, проблема в том, что p1 и p2 назначат результат обработки одной и той же переменной в конце обеих ветвей. Эта переменная будет использоваться в результате обработки и для расчета потерь.
у использоватьtf.cond (pred, fn1, fn2, name = None) для условного ветвления. Допустим, у меня есть два тензора:x, y, Каждый тензор является партией 0/1, и я хочу использовать это сжатие тензоровx < y в качестве источника дляtf.cond pred аргумент:
pred: скаляр, определяющий, возвращать ли результат fn1 или fn2.
Но если я работаю с пакетами, то, похоже, мне нужно перебрать тензор источника внутри графика, создать срезы для каждого элемента в пакете и применить tf.cond для каждого элемента. Выглядит подозрительно как для меня. Почему tf.cond не принимает пакетные и только скалярные? Можете ли вы посоветовать, как правильно использовать его в пакетном режиме?