Package org.nd4j.linalg.api.ops.impl.nlp
Class SkipGramRound
- java.lang.Object
-
- org.nd4j.autodiff.functions.DifferentialFunction
-
- org.nd4j.linalg.api.ops.DynamicCustomOp
-
- org.nd4j.linalg.api.ops.impl.nlp.SkipGramRound
-
- All Implemented Interfaces:
CustomOp
public class SkipGramRound extends DynamicCustomOp
-
-
Nested Class Summary
-
Nested classes/interfaces inherited from class org.nd4j.linalg.api.ops.DynamicCustomOp
DynamicCustomOp.DynamicCustomOpsBuilder
-
-
Field Summary
-
Fields inherited from class org.nd4j.linalg.api.ops.DynamicCustomOp
axis, bArguments, dArguments, iArguments, inplaceCall, inputArguments, outputArguments, outputVariables, sArguments, tArguments
-
Fields inherited from class org.nd4j.autodiff.functions.DifferentialFunction
dimensions, extraArgs, inPlace, ownName, ownNameSetWithDefault, sameDiff, scalarValue
-
-
Constructor Summary
Constructors Constructor Description SkipGramRound()
SkipGramRound(int target, int ngStarter, @NonNull INDArray syn0, @NonNull INDArray syn1Neg, @NonNull INDArray expTable, @NonNull INDArray negTable, int nsRounds, double alpha, long randomValue, INDArray inferenceVector)
sg ns roundSkipGramRound(int target, @NonNull INDArray syn0, @NonNull INDArray syn1, @NonNull INDArray expTable, int[] indices, byte[] codes, double alpha, long randomValue, INDArray inferenceVector)
sg hs roundSkipGramRound(@NonNull INDArray target, @NonNull INDArray ngStarter, @NonNull INDArray syn0, @NonNull INDArray syn1, @NonNull INDArray syn1Neg, @NonNull INDArray expTable, @NonNull INDArray negTable, int nsRounds, @NonNull INDArray indices, @NonNull INDArray codes, @NonNull INDArray alpha, @NonNull INDArray randomValue, INDArray inferenceVector, boolean preciseMode, int numWorkers)
full constructor
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description String
opName()
This method returns op opName as string-
Methods inherited from class org.nd4j.linalg.api.ops.DynamicCustomOp
addBArgument, addDArgument, addIArgument, addIArgument, addInputArgument, addOutputArgument, addOutputsToOp, addSArgument, addTArgument, assertValidForExecution, bArgs, builder, calculateOutputShape, calculateOutputShape, clearArrays, computeArrays, configureFromArguments, dArgs, doDiff, generateFake, generateFake, getBArgument, getDescriptor, getIArgument, getInputArgument, getOutputArgument, getSArgument, getTArgument, getValue, iArgs, initFromOnnx, initFromTensorFlow, inputArguments, mappingsForFunction, numBArguments, numDArguments, numIArguments, numInputArguments, numOutputArguments, numSArguments, numTArguments, onnxName, opHash, opNum, opType, outputArguments, outputVariables, outputVariables, propertiesForFunction, removeIArgument, removeInputArgument, removeOutputArgument, removeSArgument, removeTArgument, sArgs, setInputArgument, setInputArguments, setOutputArgument, setPropertiesForFunction, setValueFor, tArgs, tensorflowName, toString, wrapFilterNull, wrapOrNull, wrapOrNull
-
Methods inherited from class org.nd4j.autodiff.functions.DifferentialFunction
arg, arg, argNames, args, attributeAdaptersForFunction, calculateOutputDataTypes, configFieldName, configureWithSameDiff, diff, dup, equals, getBooleanFromProperty, getDoubleValueFromProperty, getIntValueFromProperty, getLongValueFromProperty, getNumOutputs, getStringFromProperty, hashCode, isConfigProperties, larg, onnxNames, outputs, outputVariable, outputVariablesNames, rarg, replaceArg, setInstanceId, tensorflowNames
-
Methods inherited from class java.lang.Object
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
-
Methods inherited from interface org.nd4j.linalg.api.ops.CustomOp
isInplaceCall
-
-
-
-
Constructor Detail
-
SkipGramRound
public SkipGramRound()
-
SkipGramRound
public SkipGramRound(int target, @NonNull @NonNull INDArray syn0, @NonNull @NonNull INDArray syn1, @NonNull @NonNull INDArray expTable, int[] indices, byte[] codes, double alpha, long randomValue, INDArray inferenceVector)
sg hs round- Parameters:
target
-syn0
-syn1
-expTable
-indices
-codes
-alpha
-randomValue
-
-
SkipGramRound
public SkipGramRound(int target, int ngStarter, @NonNull @NonNull INDArray syn0, @NonNull @NonNull INDArray syn1Neg, @NonNull @NonNull INDArray expTable, @NonNull @NonNull INDArray negTable, int nsRounds, double alpha, long randomValue, INDArray inferenceVector)
sg ns round- Parameters:
target
-ngStarter
-syn0
-syn1Neg
-expTable
-negTable
-
-
SkipGramRound
public SkipGramRound(@NonNull @NonNull INDArray target, @NonNull @NonNull INDArray ngStarter, @NonNull @NonNull INDArray syn0, @NonNull @NonNull INDArray syn1, @NonNull @NonNull INDArray syn1Neg, @NonNull @NonNull INDArray expTable, @NonNull @NonNull INDArray negTable, int nsRounds, @NonNull @NonNull INDArray indices, @NonNull @NonNull INDArray codes, @NonNull @NonNull INDArray alpha, @NonNull @NonNull INDArray randomValue, INDArray inferenceVector, boolean preciseMode, int numWorkers)
full constructor
-
-
Method Detail
-
opName
public String opName()
Description copied from class:DynamicCustomOp
This method returns op opName as string- Specified by:
opName
in interfaceCustomOp
- Overrides:
opName
in classDynamicCustomOp
- Returns:
-
-