hardmax: Hardmax

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


henry090/tfaddons documentation built on April 7, 2022, 11:27 p.m.