Returns batched one-hot vectors.
hardmax(logits, name = NULL)
Arguments
logits |
A batch tensor of logit values. |
name |
Name to use when creating ops. |
Value
A batched one-hot tensor.
Details
The depth index containing the `1` is that of the maximum logit value.