TensorFlow 부울 인덱스

Question

이제 직진 해 봅시다. 현재 텐서를 인덱싱하고 0이 아닌 항을 -1로, 0을 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 

희망이 있습니다.