Class ExternalErrorsFunction
- java.lang.Object
-
- org.nd4j.autodiff.functions.DifferentialFunction
-
- org.nd4j.linalg.api.ops.DynamicCustomOp
-
- org.nd4j.linalg.api.ops.impl.layers.ExternalErrorsFunction
-
- All Implemented Interfaces:
CustomOp
public class ExternalErrorsFunction extends DynamicCustomOp
-
-
Nested Class Summary
-
Nested classes/interfaces inherited from class org.nd4j.linalg.api.ops.DynamicCustomOp
DynamicCustomOp.DynamicCustomOpsBuilder
-
-
Field Summary
Fields Modifier and Type Field Description static StringOP_NAME-
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 ExternalErrorsFunction()ExternalErrorsFunction(SameDiff sd, List<SDVariable> inputs, Map<String,INDArray> gradients)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description List<LongShapeDescriptor>calculateOutputShape()Calculate the output shape for this opList<LongShapeDescriptor>calculateOutputShape(OpContext oc)Calculate the output shape for this opvoidconfigureFromArguments()This allows a custom op to configure relevant fields from its arguments.voidconfigureWithSameDiff(SameDiff sameDiff)List<SDVariable>doDiff(List<SDVariable> f1)The actual implementation for automatic differentiation.StringgetGradPlaceholderName()voidinitFromOnnx(Onnx.NodeProto node, SameDiff initWith, Map<String,Onnx.AttributeProto> attributesForNode, Onnx.GraphProto graph)Iniitialize the function from the givenOnnx.NodeProtovoidinitFromTensorFlow(NodeDef nodeDef, SameDiff initWith, Map<String,AttrValue> attributesForNode, GraphDef graph)Initialize the function from the givenNodeDefStringonnxName()The opName of this function in onnxStringopName()This method returns op opName as stringOp.TypeopType()The type of the opSDVariable[]outputVariables(String baseName)Return the output functions for this differential function.Map<String,Object>propertiesForFunction()Returns the properties for a given functionvoidsetPropertiesForFunction(Map<String,Object> properties)StringtensorflowName()The opName of this function tensorflowStringtoString()voidupdateBeforeExecution()-
Methods inherited from class org.nd4j.linalg.api.ops.DynamicCustomOp
addBArgument, addDArgument, addIArgument, addIArgument, addInputArgument, addOutputArgument, addOutputsToOp, addSArgument, addTArgument, assertValidForExecution, bArgs, builder, clearArrays, computeArrays, dArgs, generateFake, generateFake, getBArgument, getDescriptor, getIArgument, getInputArgument, getOutputArgument, getSArgument, getTArgument, getValue, iArgs, inputArguments, mappingsForFunction, numBArguments, numDArguments, numIArguments, numInputArguments, numOutputArguments, numSArguments, numTArguments, opHash, opNum, outputArguments, outputVariables, removeIArgument, removeInputArgument, removeOutputArgument, removeSArgument, removeTArgument, sArgs, setInputArgument, setInputArguments, setOutputArgument, setValueFor, tArgs, wrapFilterNull, wrapOrNull, wrapOrNull
-
Methods inherited from class org.nd4j.autodiff.functions.DifferentialFunction
arg, arg, argNames, args, attributeAdaptersForFunction, calculateOutputDataTypes, configFieldName, 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
-
-
-
-
Field Detail
-
OP_NAME
public static final String OP_NAME
- See Also:
- Constant Field Values
-
-
Method Detail
-
getGradPlaceholderName
public String getGradPlaceholderName()
-
configureWithSameDiff
public void configureWithSameDiff(SameDiff sameDiff)
- Overrides:
configureWithSameDiffin classDifferentialFunction
-
propertiesForFunction
public Map<String,Object> propertiesForFunction()
Description copied from class:DifferentialFunctionReturns the properties for a given function- Overrides:
propertiesForFunctionin classDynamicCustomOp- Returns:
-
configureFromArguments
public void configureFromArguments()
Description copied from interface:CustomOpThis allows a custom op to configure relevant fields from its arguments. This is needed when ops are created via reflection for things like model import.- Specified by:
configureFromArgumentsin interfaceCustomOp- Overrides:
configureFromArgumentsin classDynamicCustomOp
-
setPropertiesForFunction
public void setPropertiesForFunction(Map<String,Object> properties)
- Overrides:
setPropertiesForFunctionin classDynamicCustomOp
-
outputVariables
public SDVariable[] outputVariables(String baseName)
Description copied from class:DifferentialFunctionReturn the output functions for this differential function.- Overrides:
outputVariablesin classDynamicCustomOp- Returns:
-
doDiff
public List<SDVariable> doDiff(List<SDVariable> f1)
Description copied from class:DifferentialFunctionThe actual implementation for automatic differentiation.- Overrides:
doDiffin classDynamicCustomOp- Returns:
-
updateBeforeExecution
public void updateBeforeExecution()
-
initFromTensorFlow
public void initFromTensorFlow(NodeDef nodeDef, SameDiff initWith, Map<String,AttrValue> attributesForNode, GraphDef graph)
Description copied from class:DifferentialFunctionInitialize the function from the givenNodeDef- Overrides:
initFromTensorFlowin classDynamicCustomOp
-
initFromOnnx
public void initFromOnnx(Onnx.NodeProto node, SameDiff initWith, Map<String,Onnx.AttributeProto> attributesForNode, Onnx.GraphProto graph)
Description copied from class:DifferentialFunctionIniitialize the function from the givenOnnx.NodeProto- Overrides:
initFromOnnxin classDynamicCustomOp
-
onnxName
public String onnxName()
Description copied from class:DifferentialFunctionThe opName of this function in onnx- Overrides:
onnxNamein classDynamicCustomOp- Returns:
-
tensorflowName
public String tensorflowName()
Description copied from class:DifferentialFunctionThe opName of this function tensorflow- Overrides:
tensorflowNamein 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:
-
toString
public String toString()
- Overrides:
toStringin classDynamicCustomOp
-
calculateOutputShape
public List<LongShapeDescriptor> calculateOutputShape()
Description copied from class:DifferentialFunctionCalculate the output shape for this op- Specified by:
calculateOutputShapein interfaceCustomOp- Overrides:
calculateOutputShapein classDynamicCustomOp- Returns:
- List of output shape descriptors
-
calculateOutputShape
public List<LongShapeDescriptor> calculateOutputShape(OpContext oc)
Description copied from interface:CustomOpCalculate the output shape for this op- Specified by:
calculateOutputShapein interfaceCustomOp- Overrides:
calculateOutputShapein classDynamicCustomOp- Returns:
- Output array shapes
-
opType
public Op.Type opType()
Description copied from class:DifferentialFunctionThe type of the op- Overrides:
opTypein classDynamicCustomOp- Returns:
-
-