org.tensorflow.op.core

Class ArgMax<V extends Number>

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.core.ArgMax<V>

Type Parameters:

V - data type for output() output

All Implemented Interfaces:

Op, Operand<V>

@Operator
public final class ArgMax<V extends Number>
extends PrimitiveOp
implements Operand<V>

Returns the index with the largest value across dimensions of a tensor.

Note that in case of ties the identity of the return value is not guaranteed.

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method and Description
Output<V>	asOutput() Returns the symbolic handle of a tensor.
static <T,U extends Number> ArgMax<Long>	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<V>	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.
Output<V>	output()

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Method Detail

create

public static <V extends Number,T,U extends Number> ArgMax<V> 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.

Parameters:

scope - current graph scope

input -

dimension - int32 or int64, must be in the range `[-rank(input), rank(input))`. Describes which dimension of the input Tensor to reduce across. For vectors, use dimension = 0.

outputType -

Returns:

a new instance of ArgMax

create

public static <T,U extends Number> ArgMax<Long> 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.

Parameters:

scope - current graph scope

input -

dimension - int32 or int64, must be in the range `[-rank(input), rank(input))`. Describes which dimension of the input Tensor to reduce across. For vectors, use dimension = 0.

Returns:

a new instance of ArgMax

output

public Output<V> output()

asOutput

public Output<V> asOutput()

Description copied from interface: Operand

Returns the symbolic handle of a tensor.

Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.

Specified by:

asOutput in interface Operand<V extends Number>

See Also:

OperationBuilder.addInput(Output)