2017-09-04 6 views
1

テンソル・ランク1があります。これは次のようになります。[-1,2,3,-2,5]今、要素の絶対値に定数を追加したい、要素が負の場合要素が正の場合、何も起こりません。私のようなスカラーでこれを行う方法を知ってテンソル・ランク1の要素に数値を追加するテンソル・フローで条件が満たされた場合は、

:さらに

res = tf.cond(tensor < 0,\ 
lambda: tf.add(tf.constant(m.pi),\ 
tf.abs(tensor)),lambda: tf.constant(tensor) 

は、私はここフィボナッチ例のように、tf.scanとテンソルを反復処理する方法を知っている:

elems = np.array([1, 0, 0, 0, 0, 0]) 
initializer = (np.array(0), np.array(1)) 
fibonaccis = scan(lambda a, _: (a[1], a[0] + a[1]), elems, initializer) 

しかし、 tf.conditiontf.scanを組み合わせるにはどうすればよいですか?

答えて

1

あなただけ使用することができますtf.where

a = tf.Variable([-1,2,3,-2,5]) 
b = tf.where(tf.less(a, 0), tf.abs(a)+tf.constant(m.pi), a) 
関連する問題