public interface BroadcastOp extends Op
Modifier and Type | Method and Description |
---|---|
INDArray |
dimensions() |
int[] |
getDimension()
Dimension to do the vector op along.
|
Op.Type |
getOpType() |
void |
setDimension(int... dimension)
Set the dimension for the vector op.
|
boolean |
validateDataTypes(boolean experimentalOp) |
clearArrays, extraArgs, extraArgsBuff, extraArgsDataBuff, opName, opNum, setExtraArgs, setX, setY, setZ, toCustomOp, x, y, z
int[] getDimension()
void setDimension(int... dimension)
boolean validateDataTypes(boolean experimentalOp)
Op.Type getOpType()
INDArray dimensions()
Copyright © 2021. All rights reserved.