public class AbsoluteDifferenceLossBp extends BaseLossBp
DynamicCustomOp.DynamicCustomOpsBuilder
lossReduce
axis, bArguments, iArguments, inplaceCall, inputArguments, outputArguments, outputVariables, tArguments
dimensions, extraArgs, inPlace, sameDiff, scalarValue
Constructor and Description |
---|
AbsoluteDifferenceLossBp() |
AbsoluteDifferenceLossBp(SameDiff sameDiff,
LossReduce lossReduce,
SDVariable predictions,
SDVariable weights,
SDVariable labels) |
Modifier and Type | Method and Description |
---|---|
List<SDVariable> |
doDiff(List<SDVariable> grad)
The actual implementation for automatic differentiation.
|
String |
opName()
This method returns op opName as string
|
addArgs, calculateOutputDataTypes, getNumOutputs
addBArgument, addIArgument, addIArgument, addInputArgument, addOutputArgument, addTArgument, assertValidForExecution, bArgs, builder, calculateOutputShape, clearArrays, getBArgument, getDescriptor, getIArgument, getInputArgument, getOutputArgument, getTArgument, iArgs, initFromOnnx, initFromTensorFlow, inputArguments, numBArguments, numIArguments, numInputArguments, numOutputArguments, numTArguments, onnxName, opHash, opNum, opType, outputArguments, outputVariables, outputVariables, removeIArgument, removeInputArgument, removeOutputArgument, removeTArgument, setInputArgument, setInputArguments, setOutputArgument, tArgs, tensorflowName, toString, wrapFilterNull, wrapOrNull
arg, arg, argNames, args, attributeAdaptersForFunction, configFieldName, diff, dup, equals, f, getValue, hashCode, isConfigProperties, larg, mappingsForFunction, onnxNames, outputVariable, outputVariablesNames, propertiesForFunction, rarg, replaceArg, setInstanceId, setPropertiesForFunction, setValueFor, tensorflowNames
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
isInplaceCall
public AbsoluteDifferenceLossBp(SameDiff sameDiff, LossReduce lossReduce, SDVariable predictions, SDVariable weights, SDVariable labels)
public AbsoluteDifferenceLossBp()
public String opName()
DynamicCustomOp
opName
in interface CustomOp
opName
in class BaseLossBp
public List<SDVariable> doDiff(List<SDVariable> grad)
DifferentialFunction
doDiff
in class BaseLossBp
Copyright © 2019. All rights reserved.