【问题标题】: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))
    

    【讨论】:

    • 感谢您的解释!
    猜你喜欢
    • 2012-05-14
    • 2018-05-31
    • 1970-01-01
    • 1970-01-01
    • 2017-11-14
    • 2019-05-24
    • 2017-04-07
    • 2012-04-27
    • 2019-12-24
    相关资源
    最近更新 更多