public class AmsGradUpdater extends DynamicCustomOp
DynamicCustomOp.DynamicCustomOpsBuilder
axis, bArguments, dArguments, iArguments, inplaceCall, inputArguments, outputArguments, outputVariables, tArguments
dimensions, extraArgs, inPlace, ownName, ownNameSetWithDefault, sameDiff, scalarValue
Constructor and Description |
---|
AmsGradUpdater() |
AmsGradUpdater(@NonNull INDArray gradients,
@NonNull INDArray stateV,
@NonNull INDArray stateM,
@NonNull INDArray stateH,
double lr,
double beta1,
double beta2,
double epsilon,
int iteration) |
AmsGradUpdater(@NonNull INDArray gradients,
@NonNull INDArray stateV,
@NonNull INDArray stateM,
@NonNull INDArray stateH,
@NonNull INDArray updates,
@NonNull INDArray updatedStateV,
@NonNull INDArray updatedStateM,
@NonNull INDArray updatedStateH,
double lr,
double beta1,
double beta2,
double epsilon,
int iteration) |
Modifier and Type | Method and Description |
---|---|
String |
opName()
This method returns op opName as string
|
addBArgument, addDArgument, addIArgument, addIArgument, addInputArgument, addOutputArgument, addTArgument, assertValidForExecution, bArgs, builder, calculateOutputShape, calculateOutputShape, clearArrays, dArgs, doDiff, 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, calculateOutputDataTypes, configFieldName, diff, dup, equals, getNumOutputs, 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 AmsGradUpdater()
public AmsGradUpdater(@NonNull @NonNull INDArray gradients, @NonNull @NonNull INDArray stateV, @NonNull @NonNull INDArray stateM, @NonNull @NonNull INDArray stateH, double lr, double beta1, double beta2, double epsilon, int iteration)
public AmsGradUpdater(@NonNull @NonNull INDArray gradients, @NonNull @NonNull INDArray stateV, @NonNull @NonNull INDArray stateM, @NonNull @NonNull INDArray stateH, @NonNull @NonNull INDArray updates, @NonNull @NonNull INDArray updatedStateV, @NonNull @NonNull INDArray updatedStateM, @NonNull @NonNull INDArray updatedStateH, double lr, double beta1, double beta2, double epsilon, int iteration)
public String opName()
DynamicCustomOp
opName
in interface CustomOp
opName
in class DynamicCustomOp
Copyright © 2021. All rights reserved.