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