TensorFlow grafiğinde koşul nasıl eklenir?

Question

kodu aşağıdakilere sahip en I diyelim:

x = tf.placeholder("float32", shape=[None, ins_size**2*3], name = "x_input") 
condition = tf.placeholder("int32", shape=[1, 1], name = "condition") 
W = tf.Variable(tf.zeros([ins_size**2*3,label_option]), name = "weights") 
b = tf.Variable(tf.zeros([label_option]), name = "bias") 

if condition > 0: 
    y = tf.nn.softmax(tf.matmul(x, W) + b) 
else: 
    y = tf.nn.softmax(tf.matmul(x, W) - b) 

hesaplamada if deyimi işi (ben öyle düşünmüyorum) istiyorsunuz? Değilse, TensorFlow hesaplama grafiğine bir if ifadesini nasıl ekleyebilirim?

Answer 1

if ifadesinin burada çalışmadığından emin olun, çünkü durum grafik yapım süresinde değerlendirilmiştir, oysa tahminen koşulların çalışma zamanında yer tutucusuna verilen değere bağlı olmasını istersiniz. (condition > 0"truthy" in Python bir Tensor, olarak değerlendirildiğinden, Aslında, her zaman, birinci dal sürecektir.)

koşullu kontrol akışını desteklemek için, TensorFlow bir bağlı olarak bir, iki dallarını değerlendirir tf.cond() operatörü içerir boole durumu.

x = tf.placeholder(tf.float32, shape=[None, ins_size**2*3], name="x_input") 
condition = tf.placeholder(tf.int32, shape=[], name="condition") 
W = tf.Variable(tf.zeros([ins_size**2 * 3, label_option]), name="weights") 
b = tf.Variable(tf.zeros([label_option]), name="bias") 

y = tf.cond(condition > 0, lambda: tf.matmul(x, W) + b, lambda: tf.matmul(x, W) - b)