Learn R Programming

tfaddons (version 0.10.0)

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

Value

A batched one-hot tensor.

Details

The depth index containing the `1` is that of the maximum logit value.