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 INDArraydimensions()int[]getDimension()Dimension to do the vector op along.Op.TypegetOpType()voidsetDimension(int... dimension)Set the dimension for the vector op.booleanvalidateDataTypes(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()
-
-