public class NadamUpdater extends DynamicCustomOp
DynamicCustomOp.DynamicCustomOpsBuilder
axis, bArguments, dArguments, iArguments, inplaceCall, inputArguments, outputArguments, outputVariables, tArguments
dimensions, extraArgs, inPlace, ownName, ownNameSetWithDefault, sameDiff, scalarValue
Constructor and Description |
---|
NadamUpdater() |
NadamUpdater(@NonNull INDArray gradients,
@NonNull INDArray stateV,
@NonNull INDArray stateM,
double lr,
double beta1,
double beta2,
double epsilon,
int iteration) |
NadamUpdater(@NonNull INDArray gradients,
@NonNull INDArray stateV,
@NonNull INDArray stateM,
@NonNull INDArray updates,
@NonNull INDArray updatedStateV,
@NonNull INDArray updatedStateM,
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 NadamUpdater()
public NadamUpdater(@NonNull @NonNull INDArray gradients, @NonNull @NonNull INDArray stateV, @NonNull @NonNull INDArray stateM, double lr, double beta1, double beta2, double epsilon, int iteration)
public NadamUpdater(@NonNull @NonNull INDArray gradients, @NonNull @NonNull INDArray stateV, @NonNull @NonNull INDArray stateM, @NonNull @NonNull INDArray updates, @NonNull @NonNull INDArray updatedStateV, @NonNull @NonNull INDArray updatedStateM, 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.