View source on GitHub
|
Returns batched one-hot vectors.
tf.contrib.seq2seq.hardmax(
logits,
name=None
)
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.