public abstract class BaseBroadcastOp extends BaseOp implements BroadcastOp
Modifier and Type | Field and Description |
---|---|
protected int[] |
dimension |
extraArgs, extraArgz, n, numProcessed, passThrough, x, y, z
Constructor and Description |
---|
BaseBroadcastOp(INDArray x,
INDArray y,
INDArray z,
int... dimension) |
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.
|
exec, exec, extraArgs, extraArgsBuff, extraArgsDataBuff, getOpType, init, isExecSpecial, isPassThrough, n, numProcessed, setN, setX, setY, setZ, toString, x, y, z
clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait
exec, exec, extraArgs, extraArgsBuff, extraArgsDataBuff, init, isExecSpecial, isPassThrough, n, name, numProcessed, op, op, op, op, op, op, op, op, opNum, setExtraArgs, setN, setX, setY, setZ, x, y, z
public int broadcastLength()
BroadcastOp
broadcastLength
in interface BroadcastOp
public int[] broadcastShape()
BroadcastOp
broadcastShape
in interface BroadcastOp
public int[] getDimension()
BroadcastOp
getDimension
in interface BroadcastOp
public void setDimension(int... dimension)
BroadcastOp
setDimension
in interface BroadcastOp
public Op opForDimension(int index, int dimension)
Op
opForDimension
in interface Op
index
- the index of the op to iterate overdimension
- the dimension to ge the input forpublic Op opForDimension(int index, int... dimension)
Op
opForDimension
in interface Op
index
- the index of the op to iterate overdimension
- the dimension to ge the input forCopyright © 2017. All rights reserved.