Class SqrtM
- java.lang.Object
-
- org.nd4j.autodiff.functions.DifferentialFunction
-
- org.nd4j.linalg.api.ops.DynamicCustomOp
-
- org.nd4j.linalg.api.ops.impl.transforms.floating.SqrtM
-
- All Implemented Interfaces:
CustomOp
public class SqrtM extends DynamicCustomOp
-
-
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.DynamicCustomOp
axis, bArguments, dArguments, iArguments, inplaceCall, inputArguments, outputArguments, outputVariables, sArguments, tArguments
-
Fields inherited from class org.nd4j.autodiff.functions.DifferentialFunction
dimensions, extraArgs, inPlace, ownName, ownNameSetWithDefault, sameDiff, scalarValue
-
-
Constructor Summary
Constructors Constructor Description SqrtM()SqrtM(SameDiff sameDiff, SDVariable arg)SqrtM(SameDiff sameDiff, SDVariable[] args)SqrtM(INDArray input)SqrtM(INDArray[] inputs, INDArray[] outputs)SqrtM(INDArray[] inputs, INDArray[] outputs, List<Double> tArguments, int[] iArguments)SqrtM(INDArray[] inputs, INDArray[] outputs, List<Double> tArguments, List<Integer> iArguments)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description List<SDVariable>doDiff(List<SDVariable> f1)The actual implementation for automatic differentiation.StringopName()This method returns op opName as string-
Methods inherited from class org.nd4j.linalg.api.ops.DynamicCustomOp
addBArgument, addDArgument, addIArgument, addIArgument, addInputArgument, addOutputArgument, addOutputsToOp, addSArgument, addTArgument, assertValidForExecution, bArgs, builder, calculateOutputShape, calculateOutputShape, clearArrays, computeArrays, configureFromArguments, dArgs, 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, tensorflowName, toString, wrapFilterNull, wrapOrNull, wrapOrNull
-
Methods inherited from class org.nd4j.autodiff.functions.DifferentialFunction
arg, arg, argNames, args, attributeAdaptersForFunction, calculateOutputDataTypes, 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
-
SqrtM
public SqrtM(SameDiff sameDiff, SDVariable arg)
-
SqrtM
public SqrtM(SameDiff sameDiff, SDVariable[] args)
-
SqrtM
public SqrtM()
-
SqrtM
public SqrtM(INDArray[] inputs, INDArray[] outputs, List<Double> tArguments, int[] iArguments)
-
SqrtM
public SqrtM(INDArray[] inputs, INDArray[] outputs, List<Double> tArguments, List<Integer> iArguments)
-
SqrtM
public SqrtM(INDArray input)
-
-
Method Detail
-
doDiff
public List<SDVariable> doDiff(List<SDVariable> f1)
Description copied from class:DifferentialFunctionThe actual implementation for automatic differentiation.- Overrides:
doDiffin classDynamicCustomOp- Returns:
-
opName
public String opName()
Description copied from class:DynamicCustomOpThis method returns op opName as string- Specified by:
opNamein interfaceCustomOp- Overrides:
opNamein classDynamicCustomOp- Returns:
-
-