public class SkipGramRound extends DynamicCustomOp
DynamicCustomOp.DynamicCustomOpsBuilder
axis, bArguments, dArguments, iArguments, inplaceCall, inputArguments, outputArguments, outputVariables, tArguments
dimensions, extraArgs, inPlace, ownName, ownNameSetWithDefault, sameDiff, scalarValue
Constructor and Description |
---|
SkipGramRound() |
SkipGramRound(@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
|
SkipGramRound(int target,
@NonNull INDArray syn0,
@NonNull INDArray syn1,
@NonNull INDArray expTable,
int[] indices,
byte[] codes,
double alpha,
long randomValue,
INDArray inferenceVector)
sg hs round
|
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 round
|
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 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)
target
- syn0
- syn1
- expTable
- indices
- codes
- alpha
- randomValue
- 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)
target
- ngStarter
- syn0
- syn1Neg
- expTable
- negTable
- 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)
public String opName()
DynamicCustomOp
opName
in interface CustomOp
opName
in class DynamicCustomOp
Copyright © 2021. All rights reserved.