| 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 |
<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 <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 © 2015–2019. All rights reserved.