Class ArgMax
-
- All Implemented Interfaces:
CustomOp
public class ArgMax extends BaseDynamicCustomIndexReduction
-
-
Nested Class Summary
-
Nested classes/interfaces inherited from class org.nd4j.linalg.api.ops.DynamicCustomOp
DynamicCustomOp.DynamicCustomOpsBuilder
-
-
Field Summary
-
Fields inherited from class org.nd4j.linalg.api.ops.impl.reduce.custom.BaseDynamicCustomReduction
dimensions, isComplex, isEmptyReduce, keepDims
-
Fields inherited from class org.nd4j.linalg.api.ops.DynamicCustomOp
axis, bArguments, dArguments, iArguments, inplaceCall, inputArguments, outputArguments, outputVariables, sArguments, tArguments
-
Fields inherited from class org.nd4j.autodiff.functions.DifferentialFunction
extraArgs, inPlace, ownName, ownNameSetWithDefault, sameDiff, scalarValue
-
-
Constructor Summary
Constructors Constructor Description ArgMax()
ArgMax(boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMax(String opName, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMax(String opName, SameDiff sameDiff, SDVariable[] args, boolean inPlace, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMax(String opName, SameDiff sameDiff, SDVariable[] args, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMax(String opName, INDArray[] inputs, INDArray[] outputs, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMax(String opName, INDArray[] inputs, INDArray[] outputs, List<Double> tArguments, int[] iArguments, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMax(String opName, INDArray[] inputs, INDArray[] outputs, List<Double> tArguments, List<Integer> iArguments, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMax(String opName, INDArray input, INDArray output, List<Double> tArguments, int[] iArguments, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMax(SameDiff sameDiff, SDVariable[] args, boolean keepDims)
ArgMax(SameDiff sameDiff, SDVariable[] args, boolean keepDims, boolean isComplex)
ArgMax(SameDiff sameDiff, SDVariable[] args, boolean inPlace, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMax(SameDiff sameDiff, SDVariable[] args, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMax(SameDiff sameDiff, SDVariable[] args, boolean keepDims, boolean isComplex, int[] dimensions)
ArgMax(SameDiff sameDiff, SDVariable[] args, boolean keepDims, int[] dimensions)
ArgMax(SameDiff sameDiff, SDVariable arg, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMax(SameDiff sd, SDVariable in, boolean keepDims, int[] dimensions)
ArgMax(INDArray arr)
ArgMax(INDArray[] inputs)
ArgMax(INDArray[] inputs, boolean keepDims, int[] dimensions)
ArgMax(INDArray[] inputs, INDArray[] outputs)
ArgMax(INDArray[] inputs, INDArray[] outputs, boolean keepDims)
ArgMax(INDArray[] inputs, INDArray[] outputs, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMax(INDArray[] input, INDArray[] output, boolean keepDims, boolean isComplex, int[] dimensions)
ArgMax(INDArray[] inputs, INDArray[] outputs, boolean keepDims, int... dimensions)
ArgMax(INDArray in, boolean keepDims, int[] dimensions)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description String
opName()
This method returns op opName as stringString
tensorflowName()
The opName of this function tensorflow-
Methods inherited from class org.nd4j.linalg.api.ops.impl.reduce.custom.BaseDynamicCustomIndexReduction
calculateOutputDataTypes, calculateOutputShape
-
Methods inherited from class org.nd4j.linalg.api.ops.impl.reduce.custom.BaseDynamicCustomReduction
addArgs
-
Methods inherited from class org.nd4j.linalg.api.ops.DynamicCustomOp
addBArgument, addDArgument, addIArgument, addIArgument, addInputArgument, addOutputArgument, addOutputsToOp, addSArgument, addTArgument, assertValidForExecution, bArgs, builder, calculateOutputShape, clearArrays, computeArrays, configureFromArguments, dArgs, doDiff, generateFake, generateFake, getBArgument, getDescriptor, getIArgument, getInputArgument, getOutputArgument, getSArgument, getTArgument, getValue, iArgs, initFromOnnx, initFromTensorFlow, inputArguments, mappingsForFunction, numBArguments, numDArguments, numIArguments, numInputArguments, numOutputArguments, numSArguments, numTArguments, onnxName, opHash, opNum, opType, outputArguments, outputVariables, outputVariables, propertiesForFunction, removeIArgument, removeInputArgument, removeOutputArgument, removeSArgument, removeTArgument, sArgs, setInputArgument, setInputArguments, setOutputArgument, setPropertiesForFunction, setValueFor, tArgs, toString, wrapFilterNull, wrapOrNull, wrapOrNull
-
Methods inherited from class org.nd4j.autodiff.functions.DifferentialFunction
arg, arg, argNames, args, attributeAdaptersForFunction, configFieldName, configureWithSameDiff, diff, dup, equals, getBooleanFromProperty, getDoubleValueFromProperty, getIntValueFromProperty, getLongValueFromProperty, getNumOutputs, getStringFromProperty, hashCode, isConfigProperties, larg, onnxNames, outputs, outputVariable, outputVariablesNames, rarg, replaceArg, setInstanceId, tensorflowNames
-
Methods inherited from class java.lang.Object
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
-
Methods inherited from interface org.nd4j.linalg.api.ops.CustomOp
isInplaceCall
-
-
-
-
Constructor Detail
-
ArgMax
public ArgMax(SameDiff sameDiff, SDVariable[] args, boolean keepDims)
-
ArgMax
public ArgMax(SameDiff sameDiff, SDVariable[] args, boolean keepDims, int[] dimensions)
-
ArgMax
public ArgMax(SameDiff sameDiff, SDVariable[] args, boolean keepDims, boolean isComplex)
-
ArgMax
public ArgMax(SameDiff sameDiff, SDVariable[] args, boolean keepDims, boolean isComplex, int[] dimensions)
-
ArgMax
public ArgMax(INDArray[] inputs)
-
ArgMax
public ArgMax(INDArray[] inputs, boolean keepDims, int[] dimensions)
-
ArgMax
public ArgMax(boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMax
public ArgMax(SameDiff sameDiff, SDVariable arg, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMax
public ArgMax(SameDiff sameDiff, SDVariable[] args, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMax
public ArgMax(String opName, SameDiff sameDiff, SDVariable[] args, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMax
public ArgMax(String opName, INDArray input, INDArray output, List<Double> tArguments, int[] iArguments, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMax
public ArgMax(String opName, INDArray[] inputs, INDArray[] outputs, List<Double> tArguments, int[] iArguments, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMax
public ArgMax(String opName, INDArray[] inputs, INDArray[] outputs, List<Double> tArguments, List<Integer> iArguments, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMax
public ArgMax(INDArray[] inputs, INDArray[] outputs, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMax
public ArgMax(String opName, INDArray[] inputs, INDArray[] outputs, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMax
public ArgMax(String opName, SameDiff sameDiff, SDVariable[] args, boolean inPlace, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMax
public ArgMax(SameDiff sameDiff, SDVariable[] args, boolean inPlace, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMax
public ArgMax(String opName, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMax
public ArgMax(INDArray[] input, INDArray[] output, boolean keepDims, boolean isComplex, int[] dimensions)
-
ArgMax
public ArgMax()
-
ArgMax
public ArgMax(SameDiff sd, SDVariable in, boolean keepDims, int[] dimensions)
-
ArgMax
public ArgMax(INDArray in, boolean keepDims, int[] dimensions)
-
ArgMax
public ArgMax(INDArray arr)
-
-
Method Detail
-
opName
public String opName()
Description copied from class:DynamicCustomOp
This method returns op opName as string- Specified by:
opName
in interfaceCustomOp
- Specified by:
opName
in classBaseDynamicCustomReduction
- Returns:
-
tensorflowName
public String tensorflowName()
Description copied from class:DifferentialFunction
The opName of this function tensorflow- Overrides:
tensorflowName
in classDynamicCustomOp
- Returns:
-
-