public class CosineSimilarity extends BaseReduce3Op
Modifier and Type | Field and Description |
---|---|
static String |
OP_NAME |
isComplex, isEmptyReduce, keepDims
dimensionz, extraArgz, x, xVertexId, y, yVertexId, z, zVertexId
dimensions, extraArgs, inPlace, sameDiff, scalarValue
Constructor and Description |
---|
CosineSimilarity() |
CosineSimilarity(INDArray x,
INDArray y,
boolean allDistances,
int... dimension) |
CosineSimilarity(INDArray x,
INDArray y,
INDArray z) |
CosineSimilarity(INDArray x,
INDArray y,
INDArray z,
boolean keepDims,
boolean allDistances,
int... dimensions) |
CosineSimilarity(INDArray x,
INDArray y,
INDArray z,
boolean allDistances,
int... dimension) |
CosineSimilarity(INDArray x,
INDArray y,
INDArray z,
int... dimensions) |
CosineSimilarity(INDArray x,
INDArray y,
int... dimensions) |
CosineSimilarity(SameDiff sameDiff,
SDVariable i_v,
int[] dimensions) |
CosineSimilarity(SameDiff sameDiff,
SDVariable i_v,
SDVariable i_v2,
int[] dimensions) |
Modifier and Type | Method and Description |
---|---|
List<SDVariable> |
doDiff(List<SDVariable> i_v1)
The actual implementation for automatic differentiation.
|
static List<SDVariable> |
doDiff(SameDiff sameDiff,
SDVariable x,
SDVariable y,
SDVariable gradOut,
boolean keepDims,
int... dimensions) |
String |
opName()
The name of the op
|
int |
opNum()
The number of the op (mainly for old legacy XYZ ops
like
Op ) |
calculateOutputDataTypes, getOpType, onnxName, opType, resultType, tensorflowName
calculateOutputShape, calculateOutputShape, resultType, validateDataTypes
hasReductionIndices, initFromOnnx, initFromTensorFlow, isComplexAccumulation, isKeepDims, noOp, setDimensions
clearArrays, defineDimensions, dimensions, equals, extraArgs, extraArgsBuff, extraArgsDataBuff, getFinalResult, getInputArgument, getNumOutputs, getOpType, hashCode, outputVariables, setX, setY, setZ, toCustomOp, toString, x, y, z
arg, arg, argNames, args, attributeAdaptersForFunction, configFieldName, diff, dup, getValue, isConfigProperties, larg, mappingsForFunction, onnxNames, outputs, outputVariable, outputVariables, outputVariablesNames, propertiesForFunction, rarg, replaceArg, setInstanceId, setPropertiesForFunction, setValueFor, tensorflowNames
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
dimensions, getFinalResult, isComplexAccumulation, isKeepDims, noOp, setDimensions
clearArrays, extraArgs, extraArgsBuff, extraArgsDataBuff, setExtraArgs, setX, setY, setZ, toCustomOp, x, y, z
public static final String OP_NAME
public CosineSimilarity(SameDiff sameDiff, SDVariable i_v, int[] dimensions)
public CosineSimilarity(SameDiff sameDiff, SDVariable i_v, SDVariable i_v2, int[] dimensions)
public CosineSimilarity()
public CosineSimilarity(INDArray x, INDArray y, INDArray z, boolean allDistances, int... dimension)
public CosineSimilarity(INDArray x, INDArray y, boolean allDistances, int... dimension)
public int opNum()
DifferentialFunction
Op
)opNum
in interface Op
opNum
in class DifferentialFunction
public String opName()
DifferentialFunction
opName
in interface Op
opName
in class DifferentialFunction
public List<SDVariable> doDiff(List<SDVariable> i_v1)
DifferentialFunction
doDiff
in class DifferentialFunction
public static List<SDVariable> doDiff(SameDiff sameDiff, SDVariable x, SDVariable y, SDVariable gradOut, boolean keepDims, int... dimensions)
Copyright © 2020. All rights reserved.