hardmax: Hardmax

Description Usage Arguments Details Value

Description

Returns batched one-hot vectors.

Usage

1
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.


tfaddons documentation built on July 2, 2020, 2:12 a.m.