org.nd4j.linalg.api.ops

Class BaseBroadcastOp

java.lang.Object

org.nd4j.linalg.api.ops.BaseOp

org.nd4j.linalg.api.ops.BaseBroadcastOp

All Implemented Interfaces:

BroadcastOp, Op

Direct Known Subclasses:

BroadcastAddOp, BroadcastCopyOp, BroadcastDivOp, BroadcastMulOp, BroadcastRDivOp, BroadcastRSubOp, BroadcastSubOp

public abstract class BaseBroadcastOp
extends BaseOp
implements BroadcastOp

Field Summary

Fields

Modifier and Type	Field and Description
protected int[]	dimension

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

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

Constructor Summary

Constructors

Constructor and Description
BaseBroadcastOp(INDArray x, INDArray y, INDArray z, int... dimension)

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
int[]	getDimension() Dimension to do the vector op along.
Op	opForDimension(int index, int... dimension) A copy of this operation for a particular dimension of the input
Op	opForDimension(int index, int dimension) A copy of this operation for a particular dimension of the input
void	setDimension(int... dimension) Set the dimension for the vector op.

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

exec, exec, extraArgs, extraArgsBuff, extraArgsDataBuff, init, isExecSpecial, 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, extraArgsBuff, extraArgsDataBuff, init, isExecSpecial, isPassThrough, n, name, numProcessed, op, op, op, op, op, op, op, op, opNum, setN, setX, setY, setZ, x, y, z

Field Detail

dimension

protected int[] dimension

Constructor Detail

BaseBroadcastOp

public BaseBroadcastOp(INDArray x,
                       INDArray y,
                       INDArray z,
                       int... dimension)

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:

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

opForDimension

public Op opForDimension(int index,
                         int dimension)

Description copied from interface: Op

A copy of this operation for a particular dimension of the input

Specified by:

opForDimension in interface Op

Parameters:

index - the index of the op to iterate over

dimension - the dimension to ge the input for

Returns:

the operation for that dimension

opForDimension

public Op opForDimension(int index,
                         int... dimension)

Description copied from interface: Op

A copy of this operation for a particular dimension of the input

Specified by:

opForDimension in interface Op

Parameters:

index - the index of the op to iterate over

dimension - the dimension to ge the input for

Returns:

the operation for that dimension