Imported Upstream version 1.21.0
[platform/core/ml/nnfw.git] / res / TensorFlowPythonExamples / examples / matrix_diag / __init__.py
1 import tensorflow as tf
2
3 tf.compat.v1.disable_eager_execution()
4
5 in_ = tf.compat.v1.placeholder(dtype=tf.float32, shape=(4, 4), name="Hole")
6 op_ = tf.compat.v1.matrix_diag(in_)