Description
Returns batched one-hot vectors.
Usage
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.