public class WeightedCrossEntropyLoss extends DynamicCustomOp
DynamicCustomOp.DynamicCustomOpsBuilder, DynamicCustomOp.SameDiffBuilder
iArguments, inplaceCall, inputArguments, outputArguments, outputVariables, tArguments
dimensions, extraArgs, inPlace, sameDiff, scalarValue
Constructor and Description |
---|
WeightedCrossEntropyLoss(SameDiff sameDiff,
SDVariable targets,
SDVariable inputs,
SDVariable weights) |
Modifier and Type | Method and Description |
---|---|
String |
onnxName()
The opName of this function in onnx
|
String |
opName()
This method returns op opName as string
|
Op.Type |
opType()
The type of the op
|
String |
tensorflowName()
The opName of this function tensorflow
|
addIArgument, addIArgument, addInputArgument, addOutputArgument, addTArgument, asProperties, assertValidForExecution, builder, calculateOutputShape, doDiff, getDescriptor, getIArgument, getInputArgument, getOutputArgument, getTArgument, iArgs, initFromOnnx, initFromTensorFlow, inputArguments, numIArguments, numInputArguments, numOutputArguments, numTArguments, opHash, opNum, outputArguments, outputVariables, outputVariables, populateInputsAndOutputsFromSameDiff, removeIArgument, removeInputArgument, removeOutputArgument, removeTArgument, sameDiffBuilder, setInputArgument, setOutputArgument, tArgs, toString
arg, arg, argNames, args, attributeAdaptersForFunction, configFieldName, diff, dup, equals, f, getNumOutputs, getValue, hashCode, hasPlaceHolderInputs, isConfigProperties, larg, mappingsForFunction, onnxNames, outputVariable, outputVariablesNames, propertiesForFunction, rarg, resolvePropertiesFromSameDiffBeforeExecution, setInstanceId, setValueFor, tensorflowNames
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
isInplaceCall
public WeightedCrossEntropyLoss(SameDiff sameDiff, SDVariable targets, SDVariable inputs, SDVariable weights)
public String opName()
DynamicCustomOp
opName
in interface CustomOp
opName
in class DynamicCustomOp
public String onnxName()
DifferentialFunction
onnxName
in class DynamicCustomOp
public String tensorflowName()
DifferentialFunction
tensorflowName
in class DynamicCustomOp
public Op.Type opType()
DifferentialFunction
opType
in class DynamicCustomOp
Copyright © 2018. All rights reserved.