diff options
Diffstat (limited to 'res/TensorFlowPythonExamples/examples/matrix_band_part')
-rw-r--r-- | res/TensorFlowPythonExamples/examples/matrix_band_part/__init__.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/res/TensorFlowPythonExamples/examples/matrix_band_part/__init__.py b/res/TensorFlowPythonExamples/examples/matrix_band_part/__init__.py index 43d4d8754..a708f35c4 100644 --- a/res/TensorFlowPythonExamples/examples/matrix_band_part/__init__.py +++ b/res/TensorFlowPythonExamples/examples/matrix_band_part/__init__.py @@ -1,4 +1,6 @@ import tensorflow as tf +tf.compat.v1.disable_eager_execution() + in_ = tf.compat.v1.placeholder(dtype=tf.float32, shape=(4, 4), name="Hole") op_ = tf.compat.v1.matrix_band_part(in_, 1, -1) |