Class ArgAmax
- java.lang.Object
-
- org.nd4j.autodiff.functions.DifferentialFunction
-
- org.nd4j.linalg.api.ops.DynamicCustomOp
-
- org.nd4j.linalg.api.ops.impl.reduce.custom.BaseDynamicCustomReduction
-
- org.nd4j.linalg.api.ops.impl.reduce.custom.BaseDynamicCustomIndexReduction
-
- org.nd4j.linalg.api.ops.impl.indexaccum.custom.ArgAmax
-
- All Implemented Interfaces:
CustomOp
public class ArgAmax 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 ArgAmax()
ArgAmax(SameDiff sameDiff, SDVariable[] args, boolean keepDims)
ArgAmax(SameDiff sameDiff, SDVariable[] args, boolean keepDims, int[] dimensions)
ArgAmax(INDArray[] inputs)
ArgAmax(INDArray[] inputs, int[] dim)
ArgAmax(INDArray[] inputs, INDArray[] outputs)
ArgAmax(INDArray[] inputs, INDArray[] outputs, boolean keepDims)
ArgAmax(INDArray[] inputs, INDArray[] outputs, 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
-
ArgAmax
public ArgAmax(SameDiff sameDiff, SDVariable[] args, boolean keepDims)
-
ArgAmax
public ArgAmax(SameDiff sameDiff, SDVariable[] args, boolean keepDims, int[] dimensions)
-
ArgAmax
public ArgAmax(INDArray[] inputs)
-
ArgAmax
public ArgAmax()
-
ArgAmax
public ArgAmax(INDArray[] inputs, int[] dim)
-
-
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:
-
-