Package | Description |
---|---|
org.tensorflow.op | |
org.tensorflow.op.math |
Modifier and Type | Method and Description |
---|---|
<T,U extends Number> |
MathOps.argMax(Operand<T> input,
Operand<U> dimension)
Builds an
ArgMax operation |
<T,U extends Number> |
MathOps.argMax(Operand<T> input,
Operand<U> dimension)
Builds an
ArgMax operation |
<V extends Number,T,U extends Number> |
MathOps.argMax(Operand<T> input,
Operand<U> dimension,
Class<V> outputType)
Builds an
ArgMax operation |
<V extends Number,T,U extends Number> |
MathOps.argMax(Operand<T> input,
Operand<U> dimension,
Class<V> outputType)
Builds an
ArgMax operation |
Modifier and Type | Method and Description |
---|---|
static <T,U extends Number> |
ArgMax.create(Scope scope,
Operand<T> input,
Operand<U> dimension)
Factory method to create a class wrapping a new ArgMax operation using default output types.
|
static <T,U extends Number> |
ArgMax.create(Scope scope,
Operand<T> input,
Operand<U> dimension)
Factory method to create a class wrapping a new ArgMax operation using default output types.
|
static <V extends Number,T,U extends Number> |
ArgMax.create(Scope scope,
Operand<T> input,
Operand<U> dimension,
Class<V> outputType)
Factory method to create a class wrapping a new ArgMax operation.
|
static <V extends Number,T,U extends Number> |
ArgMax.create(Scope scope,
Operand<T> input,
Operand<U> dimension,
Class<V> outputType)
Factory method to create a class wrapping a new ArgMax operation.
|
Copyright © 2022. All rights reserved.