contrib.seq2seq.hardmax
tf.contrib.seq2seq.hardmax
tf.contrib.seq2seq.hardmax
hardmax(
logits,
name=None
)
Defined in tensorflow/contrib/seq2seq/python/ops/attention_wrapper.py.
Returns batched one-hot vectors.
The depth index containing the 1 is that of the maximum logit value.
Args:
-
logits: A batch tensor of logit values. -
name: Name to use when creating ops.
Returns:
A batched one-hot tensor.
© 2017 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/contrib/seq2seq/hardmax
