【发布时间】:2018-03-24 15:53:56
【问题描述】:
我有兴趣研究 tensorflow 中的图形。在我的数据集中,节点的数量因图而异。这是有问题的,因为在我的观察中,张量流不容易让我输入形状可变的数据。
更具体地说,当数组的形状不同时,任何人都可以提出一种简单的方法来将批量二维数组提供给 tf.Estimator.estimator 吗?我一直在使用tf.estimator.inputs.numpy_input_fn 来处理固定大小的数据集,但我现在需要一种不同的方法。
【问题讨论】:
-
您可以将批次填充到固定大小吗?
-
@mikkola 是的,这可能是要走的路。我需要填充 0,然后减小张量流图中的大小,但认为这会有点低效。
标签: tensorflow tensorflow-datasets tensorflow-estimator