Package org.nd4j.linalg.api.ops
Interface BroadcastOp
-
- All Superinterfaces:
Op
- All Known Implementing Classes:
BaseBroadcastBoolOp
,BaseBroadcastOp
,BroadcastAddOp
,BroadcastAMax
,BroadcastAMin
,BroadcastCopyOp
,BroadcastDivOp
,BroadcastEqualTo
,BroadcastGradientArgs
,BroadcastGreaterThan
,BroadcastGreaterThanOrEqual
,BroadcastLessThan
,BroadcastLessThanOrEqual
,BroadcastMax
,BroadcastMin
,BroadcastMulOp
,BroadcastNotEqual
,BroadcastRDivOp
,BroadcastRSubOp
,BroadcastSubOp
public interface BroadcastOp extends Op
-
-
Method Summary
All Methods Instance Methods Abstract Methods Modifier and Type Method 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)
-
Methods inherited from interface org.nd4j.linalg.api.ops.Op
clearArrays, extraArgs, extraArgsBuff, extraArgsDataBuff, opName, opNum, setExtraArgs, setX, setY, setZ, toCustomOp, x, y, z
-
-
-
-
Method Detail
-
getDimension
int[] getDimension()
Dimension to do the vector op along. Along dimension 1 for row vector ops, along 0 for column vector ops
-
setDimension
void setDimension(int... dimension)
Set the dimension for the vector op.
-
validateDataTypes
boolean validateDataTypes(boolean experimentalOp)
-
getOpType
Op.Type getOpType()
-
dimensions
INDArray dimensions()
-
-