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.