| hardmax {tfaddons} | R Documentation |
Hardmax
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. |
Details
The depth index containing the '1' is that of the maximum logit value.
Value
A batched one-hot tensor.
[Package tfaddons version 0.10.0 Index]