| Package | Description |
|---|---|
| org.tensorflow.op | |
| org.tensorflow.op.core |
| Modifier and Type | Method and Description |
|---|---|
<T,U extends Number> |
Ops.argMax(Operand<T> input,
Operand<U> dimension)
Adds an
ArgMax operation to the graph |
<V extends Number,T,U extends Number> |
Ops.argMax(Operand<T> input,
Operand<U> dimension,
Class<V> outputType)
Adds an
ArgMax operation to the graph |
| 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 to wrap a new ArgMax operation to the graph, 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 to wrap a new ArgMax operation to the graph.
|
Copyright © 2015–2019. All rights reserved.