2

我已经在 tensorflow v2.0 和 v1.12.0（带有tf.enable_eager_execution()）上尝试过这个。显然，如果我numpy()在函数中使用下面显示的代码片段进行main()调用，它会完美运行。但是，如果我在我的估计器模型函数中使用它，model_fn(features, labels, mode, params)那么它会抱怨'Tensor' object has no attribute 'numpy'.

ndarray = np.ones([3, 3])
tensor = tf.multiply(ndarray, 42)
print(tensor)
print(tensor.numpy())

有没有其他人遇到过类似的问题？对于 tf.estimator 来说，这似乎是个大问题？

4

1 回答 1

5

它行不通。Estimator API 与图构建相关联，它不完全支持急切执行。根据官方文档。

Estimator 的调用方法将在启用急切执行时起作用。但是，model_fn 和 input_fn 并没有急切地执行

https://www.tensorflow.org/api_docs/python/tf/estimator/Estimator

TF 2.0 甚至不支持自定义估算器，仅支持预制估算器。

于 2019-03-13T16:55:31.850 回答