__init__.py « log_softmax_2 « examples « TensorFlowPythonExamples « res - platform/core/ml/nnfw - Domain: Machine Learning / ML Framework; Licenses: Apache-2.0;

index : platform/core/ml/nnfw

Domain: Machine Learning / ML Framework; Licenses: Apache-2.0;

Sangmin Seo <sangmin7.seo@samsung.com>, Sung-Jae Lee <sjlee@mail.com>

summary refs log tree commit diff

path: root/res/TensorFlowPythonExamples/examples/log_softmax_2/__init__.py

blob: 856ebd9687040a6aaac65918eedf0a068e647c02 (plain)

import tensorflow as tf

tf.compat.v1.disable_eager_execution()

in_ = tf.compat.v1.placeholder(dtype=tf.float32, shape=(2, 3, 4, 5), name="Hole")
op_ = tf.compat.v1.nn.log_softmax(in_, axis=1)