如何在TF的ops计算中使用未知维度

Question

例如＆gt;我有一个Theano代码，有效（T == Theano）：

N = input.shape[0]  # input has shape wrt TF (?, num)
mse = T.sum(T.square(ytarg - ypred)) / N

如果TF op的计算方法未知，我真的不知道如何在变量N中使用变量N.

Answer 1

在Tensorflow中，您不需要知道在图表执行期间您将处理的元素数量。您必须使用tf.reduce_*操作将此任务委派给tensorflow。

  

减少

     

TensorFlow提供了几个操作，您可以使用这些操作执行常见的数学计算，以减少张量的各个维度。

您在Theano中定义的MSE功能可以在Tensorlow中轻松定义;

mse = tf.reduce_mean(tf.pow(tf.sub(ytarg, ytarg), 2.0))