public class Cross extends DynamicCustomOp
DynamicCustomOp.DynamicCustomOpsBuilder, DynamicCustomOp.SameDiffBuilder
iArguments, inplaceCall, inputArguments, outputArguments, outputVariables, tArguments
dimensions, extraArgs, inPlace, sameDiff, scalarValue
Constructor and Description |
---|
Cross() |
Cross(INDArray a,
INDArray b,
INDArray out) |
Cross(SameDiff sameDiff,
SDVariable[] args) |
Modifier and Type | Method and Description |
---|---|
java.util.List<SDVariable> |
doDiff(java.util.List<SDVariable> gradients)
The actual implementation for automatic differentiation.
|
java.lang.String |
opName()
This method returns op opName as string
|
java.lang.String |
tensorflowName()
The opName of this function tensorflow
|
addIArgument, addIArgument, addInputArgument, addOutputArgument, addTArgument, asProperties, assertValidForExecution, builder, calculateOutputShape, getDescriptor, getIArgument, getInputArgument, getOutputArgument, getTArgument, iArgs, initFromOnnx, initFromTensorFlow, inputArguments, numIArguments, numInputArguments, numOutputArguments, numTArguments, onnxName, opHash, opNum, opType, 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 Cross()
public Cross(SameDiff sameDiff, SDVariable[] args)
public java.lang.String opName()
DynamicCustomOp
opName
in interface CustomOp
opName
in class DynamicCustomOp
public java.lang.String tensorflowName()
DifferentialFunction
tensorflowName
in class DynamicCustomOp
public java.util.List<SDVariable> doDiff(java.util.List<SDVariable> gradients)
DifferentialFunction
doDiff
in class DynamicCustomOp