diff options
Diffstat (limited to 'tensorflow/contrib/seq2seq/python/ops/helper.py')
-rw-r--r-- | tensorflow/contrib/seq2seq/python/ops/helper.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/seq2seq/python/ops/helper.py b/tensorflow/contrib/seq2seq/python/ops/helper.py index 34367db01b..616de3199c 100644 --- a/tensorflow/contrib/seq2seq/python/ops/helper.py +++ b/tensorflow/contrib/seq2seq/python/ops/helper.py @@ -486,7 +486,7 @@ class GreedyEmbeddingHelper(Helper): # Outputs are logits, use argmax to get the most probable id if not isinstance(outputs, ops.Tensor): raise TypeError("Expected outputs to be a single Tensor, got: %s" % - outputs) + type(outputs)) sample_ids = math_ops.cast( math_ops.argmax(outputs, axis=-1), dtypes.int32) return sample_ids |