public class GELU extends BaseTransformStrictOp
dimensionz, extraArgz, x, xVertexId, y, yVertexId, z, zVertexId
dimensions, extraArgs, inPlace, ownName, ownNameSetWithDefault, sameDiff, scalarValue
Constructor and Description |
---|
GELU(INDArray ndArray) |
GELU(INDArray x,
INDArray z) |
GELU(SameDiff sameDiff,
SDVariable i_v) |
GELU(SameDiff sameDiff,
SDVariable i_v,
boolean precise) |
GELU(SameDiff sameDiff,
SDVariable i_v,
boolean inPlace,
boolean precise) |
Modifier and Type | Method and Description |
---|---|
List<SDVariable> |
doDiff(List<SDVariable> i_v)
The actual implementation for automatic differentiation.
|
String |
opName()
The name of the op
|
int |
opNum()
The number of the op (mainly for old legacy XYZ ops
like
Op ) |
calculateOutputDataTypes, calculateOutputShape, calculateOutputShape, getOpType, opType, resultType, resultType, validateDataTypes
z
clearArrays, defineDimensions, dimensions, equals, extraArgs, extraArgsBuff, extraArgsDataBuff, getFinalResult, getInputArgument, getNumOutputs, getOpType, hashCode, initFromOnnx, initFromTensorFlow, onnxName, outputVariables, setX, setY, setZ, tensorflowName, toCustomOp, toString, x, y
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
clearArrays, extraArgs, extraArgsBuff, extraArgsDataBuff, setExtraArgs, setX, setY, setZ, toCustomOp, x, y, z
public GELU(SameDiff sameDiff, SDVariable i_v, boolean inPlace, boolean precise)
public GELU(SameDiff sameDiff, SDVariable i_v, boolean precise)
public GELU(SameDiff sameDiff, SDVariable i_v)
public GELU(INDArray ndArray)
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_v)
DifferentialFunction
doDiff
in class DifferentialFunction
Copyright © 2021. All rights reserved.