在兩個單獨批次之間切換張量

Question

假設x和y是兩個N-dimensional張量，其中兩個尺寸相同，第一個尺寸爲S尺寸（批量）。比方說，b爲布爾的1-dimensional張量，大小S.

我想產生Z，一個N維張量定義爲：

z[i] = b[i] ? x[i] : y[i] for i from 0 to (S-1) 

其中x[i]是指第i個(N-1)-dimensional切片的X。

這樣做最簡單的方法是什麼？我認爲tf.cond會起作用，但它只接受標量謂詞。謝謝！

Answer 1

tf.where應該工作，並且支持這種廣播。如果你發現自己想要一個批處理版本的條件執行（其中一個或兩個分支都是昂貴的計算），那就是also possible。