org.tensorflow.op.core

Class OneHot.Options

java.lang.Object

org.tensorflow.op.core.OneHot.Options

Enclosing class:

OneHot<U>

public static class OneHot.Options
extends Object

Optional attributes for OneHot

Method Summary

Modifier and Type	Method and Description
OneHot.Options	axis(Long axis)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

axis

public OneHot.Options axis(Long axis)

Parameters:

axis - The axis to fill (default: -1, a new inner-most axis).