public class CbowRound extends DynamicCustomOp
DynamicCustomOp.DynamicCustomOpsBuilder
axis, bArguments, iArguments, inplaceCall, inputArguments, outputArguments, outputVariables, tArguments
dimensions, extraArgs, inPlace, sameDiff, scalarValue
Constructor and Description |
---|
CbowRound() |
CbowRound(INDArray target,
INDArray context,
INDArray lockedWords,
INDArray ngStarter,
INDArray syn0,
INDArray syn1,
INDArray syn1Neg,
INDArray expTable,
INDArray negTable,
INDArray indices,
INDArray codes,
int nsRounds,
INDArray alpha,
INDArray nextRandom,
INDArray inferenceVector,
INDArray numLabels,
boolean trainWords,
int numWorkers)
full constructor
|
CbowRound(int target,
int[] context,
int[] lockedWords,
INDArray syn0,
INDArray syn1,
INDArray expTable,
int[] indices,
byte[] codes,
double alpha,
long nextRandom,
INDArray inferenceVector,
int numLabels)
hs round
|
CbowRound(int target,
int[] context,
int[] lockedWords,
int ngStarter,
INDArray syn0,
INDArray syn1Neg,
INDArray expTable,
INDArray negTable,
int nsRounds,
double alpha,
long nextRandom,
INDArray inferenceVector,
int numLabels)
ns round
|
Modifier and Type | Method and Description |
---|---|
String |
opName()
This method returns op opName as string
|
addBArgument, addIArgument, addIArgument, addInputArgument, addOutputArgument, addTArgument, assertValidForExecution, bArgs, builder, calculateOutputShape, doDiff, getBArgument, getDescriptor, getIArgument, getInputArgument, getOutputArgument, getTArgument, iArgs, initFromOnnx, initFromTensorFlow, inputArguments, numBArguments, numIArguments, numInputArguments, numOutputArguments, numTArguments, onnxName, opHash, opNum, opType, outputArguments, outputVariables, outputVariables, removeIArgument, removeInputArgument, removeOutputArgument, removeTArgument, setInputArgument, setInputArguments, setOutputArgument, tArgs, tensorflowName, toString, wrapFilterNull, wrapOrNull
arg, arg, argNames, args, attributeAdaptersForFunction, calculateOutputDataTypes, configFieldName, diff, dup, equals, f, getNumOutputs, getValue, hashCode, isConfigProperties, larg, mappingsForFunction, onnxNames, outputVariable, outputVariablesNames, propertiesForFunction, rarg, replaceArg, resolvePropertiesFromSameDiffBeforeExecution, setInstanceId, setPropertiesForFunction, setValueFor, tensorflowNames
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
isInplaceCall
public CbowRound()
public 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)
target
- context
- syn0
- syn1
- expTable
- alpha
- nextRandom
- inferenceVector
- public 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)
target
- context
- ngStarter
- syn0
- syn1Neg
- expTable
- negTable
- alpha
- nextRandom
- inferenceVector
- public 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)
target
- context
- ngStarter
- syn0
- syn1
- syn1Neg
- expTable
- negTable
- alpha
- nextRandom
- inferenceVector
- public String opName()
DynamicCustomOp
opName
in interface CustomOp
opName
in class DynamicCustomOp
Copyright © 2019. All rights reserved.