【问题标题】:How to use unknown dimension in ops calculations with TF如何在使用 TF 的操作计算中使用未知维度
【发布时间】:2017-01-11 23:44:12
【问题描述】:
例如 > 我有一个 Theano 代码,它可以工作 (T == Theano):
N = input.shape[0] # input has shape wrt TF (?, num)
mse = T.sum(T.square(ytarg - ypred)) / N
如果是未知维度,我真的不知道如何在 TF op 的计算中使用变量 N。
【问题讨论】:
标签:
machine-learning
tensorflow
deep-learning
theano
【解决方案1】:
在 TensorFlow 中,您不需要知道在图形执行期间要处理的元素数量。您必须使用 tf.reduce_* 操作将此任务委托给 tensorflow。
减少
TensorFlow 提供了多种操作,您可以使用它们来执行常见的数学计算,从而减少张量的各种维度。
您在 Theano 中定义的 MSE 函数可以在 Tensorlow 中轻松定义;
mse = tf.reduce_mean(tf.pow(tf.sub(ytarg, ytarg), 2.0))