public abstract class BaseScalarOp extends BaseOp implements ScalarOp
extraArgs, extraArgz, n, numProcessed, passThrough, x, xVertexId, y, yVertexId, z, zVertexId
dimensions, inPlace, sameDiff, scalarValue
Constructor and Description |
---|
BaseScalarOp() |
BaseScalarOp(INDArray x,
INDArray y,
INDArray z,
long n,
java.lang.Number num) |
BaseScalarOp(INDArray x,
java.lang.Number num) |
BaseScalarOp(SameDiff sameDiff,
SDVariable i_v,
java.lang.Number scalar) |
BaseScalarOp(SameDiff sameDiff,
SDVariable i_v,
java.lang.Number scalar,
boolean inPlace) |
BaseScalarOp(SameDiff sameDiff,
SDVariable i_v,
java.lang.Number scalar,
boolean inPlace,
java.lang.Object[] extraArgs) |
BaseScalarOp(SameDiff sameDiff,
SDVariable i_v,
java.lang.Number scalar,
java.lang.Object[] extraArgs) |
Modifier and Type | Method and Description |
---|---|
java.util.List<long[]> |
calculateOutputShape()
Calculate
the output shape for this op
|
int[] |
getDimension()
Dimension to do the vector op along.
|
Op.Type |
opType()
The type of the op
|
java.lang.Number |
scalar()
The normal scalar
|
void |
setDimension(int... dimension)
Set the dimension for the vector op.
|
void |
setScalar(java.lang.Number scalar) |
INDArray |
z()
The resulting ndarray
|
equals, exec, exec, extraArgs, extraArgsBuff, extraArgsDataBuff, getOpType, hashCode, init, initFromOnnx, initFromTensorFlow, isExecSpecial, isPassThrough, n, numProcessed, outputVariables, setN, setX, setY, setZ, toCustomOp, toString, x, y
arg, arg, argNames, args, asProperties, attributeAdaptersForFunction, configFieldName, diff, doDiff, dup, f, getNumOutputs, getValue, hasPlaceHolderInputs, isConfigProperties, larg, mappingsForFunction, onnxName, onnxNames, opName, opNum, outputVariable, outputVariables, outputVariablesNames, propertiesForFunction, rarg, resolvePropertiesFromSameDiffBeforeExecution, setInstanceId, setValueFor, tensorflowName, tensorflowNames
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
exec, exec, extraArgs, extraArgsBuff, extraArgsDataBuff, init, isExecSpecial, isPassThrough, n, numProcessed, opName, opNum, setExtraArgs, setN, setX, setY, setZ, toCustomOp, x, y
public BaseScalarOp()
public BaseScalarOp(INDArray x, java.lang.Number num)
public BaseScalarOp(SameDiff sameDiff, SDVariable i_v, java.lang.Number scalar)
public BaseScalarOp(SameDiff sameDiff, SDVariable i_v, java.lang.Number scalar, boolean inPlace)
public BaseScalarOp(SameDiff sameDiff, SDVariable i_v, java.lang.Number scalar, boolean inPlace, java.lang.Object[] extraArgs)
public BaseScalarOp(SameDiff sameDiff, SDVariable i_v, java.lang.Number scalar, java.lang.Object[] extraArgs)
public java.util.List<long[]> calculateOutputShape()
DifferentialFunction
calculateOutputShape
in class DifferentialFunction
public Op.Type opType()
DifferentialFunction
opType
in class DifferentialFunction
public void setScalar(java.lang.Number scalar)
public java.lang.Number scalar()
ScalarOp
public int[] getDimension()
BroadcastOp
getDimension
in interface BroadcastOp
public void setDimension(int... dimension)
BroadcastOp
setDimension
in interface BroadcastOp