public class MeanPairwiseSquaredErrorLossBp extends BaseLossBp
DynamicCustomOp.DynamicCustomOpsBuilder
lossReduce
axis, bArguments, dArguments, iArguments, inplaceCall, inputArguments, outputArguments, outputVariables, tArguments
dimensions, extraArgs, inPlace, ownName, ownNameSetWithDefault, sameDiff, scalarValue
Constructor and Description |
---|
MeanPairwiseSquaredErrorLossBp() |
MeanPairwiseSquaredErrorLossBp(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, addDArgument, addIArgument, addIArgument, addInputArgument, addOutputArgument, addTArgument, assertValidForExecution, bArgs, builder, calculateOutputShape, calculateOutputShape, clearArrays, dArgs, getBArgument, getDescriptor, getIArgument, getInputArgument, getOutputArgument, getTArgument, iArgs, initFromOnnx, initFromTensorFlow, inputArguments, numBArguments, numDArguments, numIArguments, numInputArguments, numOutputArguments, numTArguments, onnxName, opHash, opNum, opType, outputArguments, outputVariables, outputVariables, removeIArgument, removeInputArgument, removeOutputArgument, removeTArgument, setInputArgument, setInputArguments, setOutputArgument, tArgs, tensorflowName, toString, wrapFilterNull, wrapOrNull, wrapOrNull
arg, arg, argNames, args, attributeAdaptersForFunction, configFieldName, diff, dup, equals, getValue, hashCode, isConfigProperties, larg, mappingsForFunction, onnxNames, outputs, outputVariable, outputVariablesNames, propertiesForFunction, rarg, replaceArg, setInstanceId, setPropertiesForFunction, setValueFor, tensorflowNames
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
isInplaceCall
public MeanPairwiseSquaredErrorLossBp(SameDiff sameDiff, LossReduce lossReduce, SDVariable predictions, SDVariable weights, SDVariable labels)
public MeanPairwiseSquaredErrorLossBp()
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 © 2021. All rights reserved.