tensorflow - Tensorflow EagerTensor 是在哪里定义的？

Question

在我的代码中，我想检查返回的对象类型是否为 EagerTensor :

import tensorflow as tf
import inspect

if __name__ == '__main__':

    tf.enable_eager_execution()
    iterator = tf.data.Dataset.from_tensor_slices([[1, 2], [3, 4]]).__iter__()
    elem = iterator.next()
    print(type(elem))
    print(inspect.getmodule(elem))
    assert type(elem) == tf.python.framework.ops.EagerTensor

但结果是:

<class 'EagerTensor'>
<module 'tensorflow.python.framework.ops' from '/home/antek/anaconda3/envs/mnist_identification/lib/python3.6/site-packages/tensorflow/python/framework/ops.py'>
Traceback (most recent call last):
  File "/home/antek/.PyCharm2018.1/config/scratches/scratch_4.py", line 11, in <module>
    assert type(elem) == tf.python.framework.ops.EagerTensor
AttributeError: module 'tensorflow' has no attribute 'python'

这里: AttributeError: module 'tensorflow' has no attribute 'python'我发现 tensorflow 故意删除了它对 python 模块的引用。那么如何检查我的对象是否是 EagerTensor 实例？

Answer 1

我不确定你是否可以，但我认为你可能不需要。您已经拥有以下工具:

tf.is_tensor (以前的 tf.contrib.framework.is_tensor )将返回 True对于 EagerTensor

tf.executing_eagerly返回 True如果你是，那么，急切地执行。

我相信它们应该满足您 99% 的需求——如果它落在被忽略的那个百分比内，我很想知道您的问题。