org.nd4j.linalg.api.ops

Class BaseScalarOp

java.lang.Object

org.nd4j.linalg.api.ops.BaseOp

org.nd4j.linalg.api.ops.BaseScalarOp

All Implemented Interfaces:

BroadcastOp, Op, ScalarOp

Direct Known Subclasses:

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

public abstract class BaseScalarOp
extends BaseOp
implements ScalarOp

Base scalar operation

Author:

Adam Gibson

Field Summary

Fields

Modifier and Type	Field and Description
protected IComplexNumber	complexNumber
protected Number	num

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

extraArgs, n, numProcessed, passThrough, x, y, z

Constructor Summary

Constructors

Constructor and Description
BaseScalarOp()
BaseScalarOp(INDArray x, IComplexNumber num)
BaseScalarOp(INDArray x, INDArray y, INDArray z, int n, IComplexNumber num)
BaseScalarOp(INDArray x, INDArray y, INDArray z, int n, Number num)
BaseScalarOp(INDArray x, Number num)

Method Summary

Modifier and Type	Method and Description
int	broadcastLength() The length of the number of elements in the broadcast
int[]	broadcastShape() The shape of the element to be broadcast
IComplexNumber	complexScalar() The complex scalar
int[]	getDimension() Dimension to do the vector op along.
Number	scalar() The normal scalar
void	setDimension(int... dimension) Set the dimension for the vector op.

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

ensureProperVectors, exec, exec, extraArgs, init, isPassThrough, n, numProcessed, setN, setX, setY, setZ, toString, x, y, z

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

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

exec, exec, extraArgs, init, isPassThrough, n, name, numProcessed, op, op, op, op, op, op, op, op, opForDimension, opForDimension, setN, setX, setY, setZ, x, y, z

Field Detail

num

protected Number num

complexNumber

protected IComplexNumber complexNumber

Constructor Detail

BaseScalarOp

public BaseScalarOp()

BaseScalarOp

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

BaseScalarOp

public BaseScalarOp(INDArray x,
                    Number num)

BaseScalarOp

public BaseScalarOp(INDArray x,
                    INDArray y,
                    INDArray z,
                    int n,
                    IComplexNumber num)

BaseScalarOp

public BaseScalarOp(INDArray x,
                    IComplexNumber num)

Method Detail

broadcastLength

public int broadcastLength()

Description copied from interface: BroadcastOp

The length of the number of elements in the broadcast

Specified by:

broadcastLength in interface BroadcastOp

Returns:

broadcastShape

public int[] broadcastShape()

Description copied from interface: BroadcastOp

The shape of the element to be broadcast

Specified by:

broadcastShape in interface BroadcastOp

Returns:

scalar

public Number scalar()

Description copied from interface: ScalarOp

The normal scalar

Specified by:

scalar in interface ScalarOp

Returns:

the scalar

complexScalar

public IComplexNumber complexScalar()

Description copied from interface: ScalarOp

The complex scalar

Specified by:

complexScalar in interface ScalarOp

Returns:

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