2つの別々のバッチでのテンソル間のトグル

Question

xとyが2つのN-dimensionalテンソルであるとしましょう。両方が同じ次元で、最初の次元はサイズS（バッチサイズ）です。

x[i]はのi番目の (N-1)-dimensionalスライスを指し

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

：私は、z、のように定義N次元テンソルを生成したいのは、

S.

bは大きさの、ブール値の 1-dimensionalテンソルであるとしましょうバツ。

これを行う最も簡単な方法は何ですか？私はtf.condはうまくいくと思っていましたが、スカラー値の述語しか受け付けません。ありがとうございました！

Answer 1

tf.whereは、この種のブロードキャストをサポートします。条件付き実行のバッチバージョン（一方または両方のブランチの計算に費用がかかる）が必要な場合は、also possibleです。