org.nd4j.linalg.api.ops

Class BaseScalarOp

java.lang.Object

org.nd4j.autodiff.functions.DifferentialFunction

org.nd4j.linalg.api.ops.BaseOp

org.nd4j.linalg.api.ops.BaseScalarOp

All Implemented Interfaces:

BroadcastOp, Op, ScalarOp

Direct Known Subclasses:

ScalarAdd, ScalarDivision, ScalarEquals, ScalarFMod, ScalarGreaterThan, ScalarGreaterThanOrEqual, ScalarLessThan, ScalarLessThanOrEqual, ScalarMax, ScalarMin, ScalarMultiplication, ScalarNotEquals, ScalarRemainder, ScalarReverseDivision, ScalarReverseSubtraction, ScalarSet, ScalarSetValue, ScalarSubtraction

public abstract class BaseScalarOp
extends BaseOp
implements ScalarOp

Base scalar operation

Author:

Adam Gibson

Nested Class Summary

Nested classes/interfaces inherited from interface org.nd4j.linalg.api.ops.Op

Op.Type

Field Summary

Fields inherited from class org.nd4j.linalg.api.ops.BaseOp

extraArgs, extraArgz, n, numProcessed, passThrough, x, xVertexId, y, yVertexId, z, zVertexId

Fields inherited from class org.nd4j.autodiff.functions.DifferentialFunction

dimensions, inPlace, sameDiff, scalarValue

Constructor Summary

Constructors

Constructor and Description
BaseScalarOp()
BaseScalarOp(INDArray x, INDArray y, INDArray z, long n, Number num)
BaseScalarOp(INDArray x, Number num)
BaseScalarOp(SameDiff sameDiff, SDVariable i_v, Number scalar)
BaseScalarOp(SameDiff sameDiff, SDVariable i_v, Number scalar, boolean inPlace)
BaseScalarOp(SameDiff sameDiff, SDVariable i_v, Number scalar, boolean inPlace, Object[] extraArgs)
BaseScalarOp(SameDiff sameDiff, SDVariable i_v, Number scalar, Object[] extraArgs)

Method Summary

Modifier and Type	Method and Description
List<int[]>	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
Number	scalar() The normal scalar
void	setDimension(int... dimension) Set the dimension for the vector op.
void	setScalar(Number scalar)
INDArray	z() The resulting ndarray

Methods inherited from class org.nd4j.linalg.api.ops.BaseOp

equals, exec, exec, extraArgs, extraArgsBuff, extraArgsDataBuff, getOpType, hashCode, init, initFromOnnx, initFromTensorFlow, isExecSpecial, isPassThrough, n, numProcessed, outputVariables, setN, setX, setY, setZ, toCustomOp, toString, x, y

Methods inherited from class org.nd4j.autodiff.functions.DifferentialFunction

arg, args, asProperties, attributeAdaptersForFunction, configFieldName, diff, doDiff, dup, f, getValue, hasPlaceHolderInputs, isConfigProperties, larg, mappingsForFunction, onnxName, onnxNames, opName, opNum, outputVariables, propertiesForFunction, rarg, resolvePropertiesFromSameDiffBeforeExecution, setInstanceId, setValueFor, tensorflowName, tensorflowNames

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Methods inherited from interface org.nd4j.linalg.api.ops.Op

exec, exec, extraArgs, extraArgsBuff, extraArgsDataBuff, init, isExecSpecial, isPassThrough, n, numProcessed, opName, opNum, setExtraArgs, setN, setX, setY, setZ, toCustomOp, x, y

Constructor Detail

BaseScalarOp

public BaseScalarOp()

BaseScalarOp

public BaseScalarOp(INDArray x,
                    INDArray y,
                    INDArray z,
                    long n,
                    Number num)

BaseScalarOp

public BaseScalarOp(INDArray x,
                    Number num)

BaseScalarOp

public BaseScalarOp(SameDiff sameDiff,
                    SDVariable i_v,
                    Number scalar)

BaseScalarOp

public BaseScalarOp(SameDiff sameDiff,
                    SDVariable i_v,
                    Number scalar,
                    boolean inPlace)

BaseScalarOp

public BaseScalarOp(SameDiff sameDiff,
                    SDVariable i_v,
                    Number scalar,
                    boolean inPlace,
                    Object[] extraArgs)

BaseScalarOp

public BaseScalarOp(SameDiff sameDiff,
                    SDVariable i_v,
                    Number scalar,
                    Object[] extraArgs)

Method Detail

z

public INDArray z()

Description copied from interface: Op

The resulting ndarray

Specified by:

z in interface Op

Overrides:

z in class BaseOp

Returns:

the resulting ndarray

calculateOutputShape

public List<int[]> calculateOutputShape()

Description copied from class: DifferentialFunction

Calculate the output shape for this op

Overrides:

calculateOutputShape in class DifferentialFunction

Returns:

opType

public Op.Type opType()

Description copied from class: DifferentialFunction

The type of the op

Overrides:

opType in class DifferentialFunction

Returns:

setScalar

public void setScalar(Number scalar)

Specified by:

setScalar in interface ScalarOp

scalar

public Number scalar()

Description copied from interface: ScalarOp

The normal scalar

Specified by:

scalar in interface ScalarOp

Returns:

the scalar

getDimension

public int[] getDimension()

Description copied from interface: BroadcastOp

Dimension to do the vector op along. Along dimension 1 for row vector ops, along 0 for column vector ops

Specified by:

getDimension in interface BroadcastOp

setDimension

public void setDimension(int... dimension)

Description copied from interface: BroadcastOp

Set the dimension for the vector op.

Specified by:

setDimension in interface BroadcastOp