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