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. |