在 Tensorflow 中通过字符串选择不同的模式 [英] Select different modes by string in Tensorflow
问题描述
我正在尝试构建一个 VAE 网络,我希望模型在其中以不同的模式做不同的事情.我有三种模式:训练"、相同"和不同"以及一个名为插值(模式)的函数,它根据模式执行不同的操作.我的代码看起来像:
I am trying to build a VAE network in which I want the model to do different things in different modes. I have three modes: "train", "same" and "different" and a function named interpolation(mode) that does different things depend on the mode. My code looks like:
import tensorflow as tf
### some code here
mode = tf.placeholder(dtype = tf.string, name = "mode")
def interpolation(mode):
if mode == "train":
# do something
print("enter train mode")
elif mode == "same":
# do other things
print("enter same mode")
else:
# do other things
print("enter different mode")
# some other code here
sess.run(feed_dict = {mode: "train"})
sess.run(feed_dict = {mode: "same"})
sess.run(feed_dict = {mode: "different"})
但输出看起来像:
enter different mode
enter different mode
enter different mode
这意味着传入的模式不会改变条件.我做错了什么?如何通过字符串参数选择模式?
which means the mode that gets passed in doesn't change the condition. What have I done wrong? How do I select mode by string argument?
推荐答案
第一种方法:您可以使用原生 Tensorflow 切换案例.比如我假设你有三种情况,那么你可以这样做:
First approach: You can select a different mode by using native Tensorflow switch-case. For example, I assume you have three cases, then you can do:
import tensorflow as tf
mode = tf.placeholder(tf.string, shape=[], name="mode")
def cond1():
return tf.constant('same')
def cond2():
return tf.constant('train')
def cond3():
return tf.constant('diff')
def cond4():
return tf.constant('default')
y = tf.case({tf.equal(mode, 'same'): cond1,
tf.equal(mode, 'train'): cond2,
tf.equal(mode, 'diff'): cond3},
default=cond4, exclusive=True)
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
print(sess.run(y, feed_dict={mode: "train"}))
print(sess.run(y, feed_dict={mode: "same"}))
第二种方法:这是使用新的 AutoGraph API 执行此操作的另一种方法:
import tensorflow as tf
from tensorflow.contrib import autograph as ag
m = tf.placeholder(dtype=tf.string, name='mode')
def interpolation(mode):
if mode == "train":
return 'I am train'
elif mode == "same":
return 'I am same'
else:
return 'I am different'
cond_func = ag.to_graph(interpolation)(m)
with tf.Session() as sess:
print(sess.run(cond_func, feed_dict={m: 'same'}))
这篇关于在 Tensorflow 中通过字符串选择不同的模式的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!