TensorFlowブールインデックス

Question

直進しよう、現在、私はテンソルのインデックスを作成し、非ゼロの項を-1に、ゼロの項を1に変更したいと思います。しかし、TensorFlowでそれを行う方法はわかりません。

y_[y_ != 0].assign(-1) 
y_[y_ == 0].assign(1) 

理由はTensorFlowはブールインデックス作成をサポートしていないようです。

は、ここに私のコードです。どうすれば修正できますか？

Btw、私にとってはboolean_maskはうまくいきません。なぜなら、私はy_のスライスを返したくないからです。y_の値を変更したいだけです。

ありがとうございます！

Answer 1

条件付き割り当てにはtf.cond（）を使用できます。私は以下のサンプルコードを与えました。

import tensorflow as tf 

x_= tf.Variable(5) #non-zero variable 
y_= tf.Variable(0) #variable euqals to 0 

y_ =tf.cond(tf.equal(y_,0),lambda :y_.assign(1),lambda :y_.assign(-1)) #assign 1 if variable equals to zero else -1 
x_ =tf.cond(tf.equal(x_,0),lambda :x_.assign(1),lambda :x_.assign(-1)) #assign 1 if variable equals to zero else -1 

sess = tf.Session() 
sess.run(tf.global_variables_initializer()) 

with sess.as_default(): 
    print(y_.eval()) #prints 1 
    print(x_.eval()) #prints -1 

希望します。