【发布时间】:2019-08-27 15:07:41
【问题描述】:
我想根据某些条件更改输入或/和输出。但毕业生是无。
我怎样才能得到渐变并像下面的代码一样保持选择?
input1, input2 = ..., ...
output1, output2 = model(input1), model(input2)
input = tf.where(tf.less(output1, output2), input1, input2)
output = tf.where(tf.less(output1, output2), output1, output2)
grad, = tf.gradient(output, input)
我打印输入和输出的类型,它们的尺寸与输入1/输出1(输入2/输出2)相同。而如果我只计算 tf.gradient(output1,input1) 也没问题。它们有什么区别?
【问题讨论】:
标签: python tensorflow