Class ArgMin
-
- All Implemented Interfaces:
CustomOp
public class ArgMin 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 ArgMin()
ArgMin(boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMin(String opName, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMin(String opName, SameDiff sameDiff, SDVariable[] args, boolean inPlace, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMin(String opName, SameDiff sameDiff, SDVariable[] args, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMin(String opName, INDArray[] inputs, INDArray[] outputs, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMin(String opName, INDArray[] inputs, INDArray[] outputs, List<Double> tArguments, int[] iArguments, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMin(String opName, INDArray[] inputs, INDArray[] outputs, List<Double> tArguments, List<Integer> iArguments, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMin(String opName, INDArray input, INDArray output, List<Double> tArguments, int[] iArguments, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMin(SameDiff sameDiff, SDVariable[] args, boolean keepDims)
ArgMin(SameDiff sameDiff, SDVariable[] args, boolean keepDims, boolean isComplex)
ArgMin(SameDiff sameDiff, SDVariable[] args, boolean inPlace, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMin(SameDiff sameDiff, SDVariable[] args, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMin(SameDiff sameDiff, SDVariable[] args, boolean keepDims, boolean isComplex, int[] dimensions)
ArgMin(SameDiff sameDiff, SDVariable[] args, boolean keepDims, int[] dimensions)
ArgMin(SameDiff sameDiff, SDVariable arg, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMin(SameDiff sd, SDVariable in, boolean keepDims, int[] dimensions)
ArgMin(INDArray arr)
ArgMin(INDArray[] inputs)
ArgMin(INDArray[] inputs, boolean keepDims, int[] dimensions)
ArgMin(INDArray[] inputs, INDArray[] outputs)
ArgMin(INDArray[] inputs, INDArray[] outputs, boolean keepDims)
ArgMin(INDArray[] inputs, INDArray[] outputs, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
ArgMin(INDArray[] input, INDArray[] output, boolean keepDims, boolean isComplex, int[] dimensions)
ArgMin(INDArray[] inputs, INDArray[] outputs, boolean keepDims, int... dimensions)
ArgMin(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
-
ArgMin
public ArgMin(SameDiff sameDiff, SDVariable[] args, boolean keepDims)
-
ArgMin
public ArgMin(SameDiff sameDiff, SDVariable[] args, boolean keepDims, int[] dimensions)
-
ArgMin
public ArgMin(SameDiff sameDiff, SDVariable[] args, boolean keepDims, boolean isComplex)
-
ArgMin
public ArgMin(SameDiff sameDiff, SDVariable[] args, boolean keepDims, boolean isComplex, int[] dimensions)
-
ArgMin
public ArgMin(INDArray[] inputs)
-
ArgMin
public ArgMin(INDArray[] inputs, boolean keepDims, int[] dimensions)
-
ArgMin
public ArgMin(boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMin
public ArgMin(SameDiff sameDiff, SDVariable arg, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMin
public ArgMin(SameDiff sameDiff, SDVariable[] args, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMin
public ArgMin(String opName, SameDiff sameDiff, SDVariable[] args, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMin
public ArgMin(String opName, INDArray input, INDArray output, List<Double> tArguments, int[] iArguments, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMin
public ArgMin(String opName, INDArray[] inputs, INDArray[] outputs, List<Double> tArguments, int[] iArguments, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMin
public ArgMin(String opName, INDArray[] inputs, INDArray[] outputs, List<Double> tArguments, List<Integer> iArguments, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMin
public ArgMin(INDArray[] inputs, INDArray[] outputs, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMin
public ArgMin(String opName, INDArray[] inputs, INDArray[] outputs, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMin
public ArgMin(String opName, SameDiff sameDiff, SDVariable[] args, boolean inPlace, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMin
public ArgMin(SameDiff sameDiff, SDVariable[] args, boolean inPlace, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMin
public ArgMin(String opName, boolean keepDims, boolean isComplex, boolean isEmptyReduce, int[] dimensions)
-
ArgMin
public ArgMin(INDArray[] input, INDArray[] output, boolean keepDims, boolean isComplex, int[] dimensions)
-
ArgMin
public ArgMin()
-
ArgMin
public ArgMin(SameDiff sd, SDVariable in, boolean keepDims, int[] dimensions)
-
ArgMin
public ArgMin(INDArray in, boolean keepDims, int[] dimensions)
-
ArgMin
public ArgMin(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:
-
-