org.nd4j.linalg.api.ops

Class BaseVectorOp

java.lang.Object

org.nd4j.linalg.api.ops.BaseOp

org.nd4j.linalg.api.ops.BaseVectorOp

All Implemented Interfaces:

Op, VectorOp

Direct Known Subclasses:

VectorAddOp, VectorCopyOp, VectorDivOp, VectorMulOp, VectorRDivOp, VectorRSubOp, VectorSubOp

public abstract class BaseVectorOp
extends BaseOp
implements VectorOp

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
BaseVectorOp(INDArray x, INDArray y, INDArray z, int dimension)

Method Summary

Modifier and Type	Method and Description
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

ensureProperVectors, exec, exec, extraArgs, init, isPassThrough, n, numProcessed, 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, setX, setY, setZ, x, y, z

Field Detail

dimension

protected int dimension

Constructor Detail

BaseVectorOp

public BaseVectorOp(INDArray x,
                    INDArray y,
                    INDArray z,
                    int dimension)

Method Detail

getDimension

public int getDimension()

Description copied from interface: VectorOp

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 VectorOp

setDimension

public void setDimension(int dimension)

Description copied from interface: VectorOp

Set the dimension for the vector op.

Specified by:

setDimension in interface VectorOp

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