diff options
| -rw-r--r-- | model-integration/src/test/models/tensorflow/softmax/softmax.py | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/model-integration/src/test/models/tensorflow/softmax/softmax.py b/model-integration/src/test/models/tensorflow/softmax/softmax.py index c636af58043..aab9956f914 100644 --- a/model-integration/src/test/models/tensorflow/softmax/softmax.py +++ b/model-integration/src/test/models/tensorflow/softmax/softmax.py @@ -9,11 +9,9 @@ n_inputs = 5 n_outputs = 3 input = tf.placeholder(tf.float32, shape=(None, n_inputs), name="input") - -W = tf.Variable(tf.random.uniform([n_inputs, n_outputs]), name="weights") # or just add uniform random +W = tf.Variable(tf.random.uniform([n_inputs, n_outputs]), name="weights") b = tf.Variable(tf.random.uniform([n_outputs]), name="bias") Z = tf.matmul(input, W) + b - hidden_layer = tf.nn.relu(Z) output_layer = tf.nn.softmax(hidden_layer, name="output") |