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

Nested Class Summary

Nested classes/interfaces inherited from interface org.nd4j.linalg.api.ops.Op

Op.Type

Method Summary

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)

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()