diff options
author | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-24 13:51:51 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-24 13:51:51 -0700 |
commit | 081d9b7fa17fb9f4ea39b5ac5cc20432ae5d1756 (patch) | |
tree | e7a7b23bef03a598bac664bb98eb0f3da616a7c6 /tensorflow/examples | |
parent | cc7525d2c8e44c06685b27e77450cb2114765d72 (diff) | |
parent | 587ec10386c3a939b4dbb7ead2d05516180d95c5 (diff) |
Merge pull request #22416 from yongtang:09182018-squeeze_dims
PiperOrigin-RevId: 214323563
Diffstat (limited to 'tensorflow/examples')
-rw-r--r-- | tensorflow/examples/learn/text_classification_character_cnn.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/examples/learn/text_classification_character_cnn.py b/tensorflow/examples/learn/text_classification_character_cnn.py index afda170e2a..b8506fa8a4 100644 --- a/tensorflow/examples/learn/text_classification_character_cnn.py +++ b/tensorflow/examples/learn/text_classification_character_cnn.py @@ -74,7 +74,7 @@ def char_cnn_model(features, labels, mode): kernel_size=FILTER_SHAPE2, padding='VALID') # Max across each filter to get useful features for classification. - pool2 = tf.squeeze(tf.reduce_max(conv2, 1), squeeze_dims=[1]) + pool2 = tf.squeeze(tf.reduce_max(conv2, 1), axis=[1]) # Apply regular WX + B and classification. logits = tf.layers.dense(pool2, MAX_LABEL, activation=None) |