public class FirstIndex extends BaseIndexAccumulation
Modifier and Type | Field and Description |
---|---|
protected double |
compare |
protected Condition |
condition |
protected double |
eps |
protected int |
mode |
keepDims
dimensionz, extraArgz, x, xVertexId, y, yVertexId, z, zVertexId
dimensions, extraArgs, inPlace, sameDiff, scalarValue
Constructor and Description |
---|
FirstIndex(INDArray x,
boolean keepDims,
@NonNull Condition condition,
int... dimension) |
FirstIndex(INDArray x,
@NonNull Condition condition,
boolean keepDims,
int... dimension) |
FirstIndex(INDArray x,
@NonNull Condition condition,
double eps,
int... dimension) |
FirstIndex(INDArray x,
@NonNull Condition condition,
int... dimension) |
FirstIndex(SameDiff sameDiff,
SDVariable i_v,
boolean keepDims,
Condition condition,
int... dimensions) |
FirstIndex(SameDiff sameDiff,
SDVariable i_v,
Condition condition,
boolean keepDims,
int... dimensions) |
Modifier and Type | Method and Description |
---|---|
List<SDVariable> |
doDiff(List<SDVariable> f1)
The actual implementation for automatic differentiation.
|
String |
onnxName()
The opName of this function in onnx
|
String |
opName()
The name of the op
|
int |
opNum()
The number of the op (mainly for old legacy XYZ ops
like
Op ) |
String |
tensorflowName()
The opName of this function tensorflow
|
calculateOutputDataTypes, calculateOutputShape, calculateOutputShape, opType, validateDataTypes
clearArrays, defineDimensions, dimensions, equals, extraArgs, extraArgsBuff, extraArgsDataBuff, getFinalResult, getInputArgument, getNumOutputs, getOpType, hashCode, initFromOnnx, initFromTensorFlow, 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, isKeepDims
clearArrays, extraArgs, extraArgsBuff, extraArgsDataBuff, setExtraArgs, setX, setY, setZ, toCustomOp, x, y, z
protected Condition condition
protected double compare
protected double eps
protected int mode
public FirstIndex(SameDiff sameDiff, SDVariable i_v, Condition condition, boolean keepDims, int... dimensions)
public FirstIndex(SameDiff sameDiff, SDVariable i_v, boolean keepDims, Condition condition, int... dimensions)
public FirstIndex(INDArray x, boolean keepDims, @NonNull @NonNull Condition condition, int... dimension)
public FirstIndex(INDArray x, @NonNull @NonNull Condition condition, boolean keepDims, 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> f1)
DifferentialFunction
doDiff
in class DifferentialFunction
public String onnxName()
DifferentialFunction
public String tensorflowName()
DifferentialFunction
tensorflowName
in class BaseOp
Copyright © 2020. All rights reserved.