diff options
Diffstat (limited to 'res/TensorFlowPythonExamples/examples/gather_nd/__init__.py')
-rw-r--r-- | res/TensorFlowPythonExamples/examples/gather_nd/__init__.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/res/TensorFlowPythonExamples/examples/gather_nd/__init__.py b/res/TensorFlowPythonExamples/examples/gather_nd/__init__.py index 1ff11d568..8c0df3629 100644 --- a/res/TensorFlowPythonExamples/examples/gather_nd/__init__.py +++ b/res/TensorFlowPythonExamples/examples/gather_nd/__init__.py @@ -1,5 +1,7 @@ import tensorflow as tf +tf.compat.v1.disable_eager_execution() + param_ = tf.compat.v1.placeholder(dtype=tf.float32, shape=(2, 2, 2), name="Hole") indices_ = tf.constant([[0, 1], [1, 0]]) op_ = tf.gather_nd(param_, indices_) |