org.nd4j.linalg.factory

Class Broadcast

java.lang.Object

org.nd4j.linalg.factory.Broadcast

public class Broadcast
extends Object

Method Summary

Modifier and Type	Method and Description
static INDArray	add(INDArray x, INDArray y, INDArray z, int... dimensions) Broadcast add op.
static INDArray	amax(INDArray x, INDArray y, INDArray z, int... dimensions) Broadcast absolute max op.
static INDArray	amin(INDArray x, INDArray y, INDArray z, int... dimensions) Broadcast absolute min op.
static INDArray	copy(INDArray x, INDArray y, INDArray z, int... dimensions) Broadcast copy op.
static INDArray	div(INDArray x, INDArray y, INDArray z, int... dimensions) Broadcast divide op.
static INDArray	eq(INDArray x, INDArray y, INDArray z, int... dimensions) Broadcast equal to op.
static INDArray	gt(INDArray x, INDArray y, INDArray z, int... dimensions) Broadcast greater than op.
static INDArray	gte(INDArray x, INDArray y, INDArray z, int... dimensions) Broadcast greater than or equal to op.
static INDArray	lt(INDArray x, INDArray y, INDArray z, int... dimensions) Broadcast less than op.
static INDArray	lte(INDArray x, INDArray y, INDArray z, int... dimensions) Broadcast less than or equal to op.
static INDArray	max(INDArray x, INDArray y, INDArray z, int... dimensions) Broadcast max op.
static INDArray	min(INDArray x, INDArray y, INDArray z, int... dimensions) Broadcast min op.
static INDArray	mul(INDArray x, INDArray y, INDArray z, int... dimensions) Broadcast element-wise multiply op.
static INDArray	neq(INDArray x, INDArray y, INDArray z, int... dimensions) Broadcast not equal to op.
static INDArray	rdiv(INDArray x, INDArray y, INDArray z, int... dimensions) Broadcast reverse division op.
static INDArray	rsub(INDArray x, INDArray y, INDArray z, int... dimensions) Broadcast reverse subtraction op.
static INDArray	sub(INDArray x, INDArray y, INDArray z, int... dimensions) Broadcast subtraction op.
static void	validateBroadcastDims(INDArray x, INDArray y, INDArray z, int... dimensions) Validate the broadcast dimensions for manual broadcast ops such as BroadcastMulOp.
static void	validateShapesNoDimCase(INDArray x, INDArray y, INDArray z)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

add

public static INDArray add(INDArray x,
                           INDArray y,
                           INDArray z,
                           int... dimensions)

Broadcast add op. See: BroadcastAddOp

copy

public static INDArray copy(INDArray x,
                            INDArray y,
                            INDArray z,
                            int... dimensions)

Broadcast copy op. See: BroadcastCopyOp

div

public static INDArray div(INDArray x,
                           INDArray y,
                           INDArray z,
                           int... dimensions)

Broadcast divide op. See: BroadcastDivOp

eq

public static INDArray eq(INDArray x,
                          INDArray y,
                          INDArray z,
                          int... dimensions)

Broadcast equal to op. See: BroadcastEqualTo

gt

public static INDArray gt(INDArray x,
                          INDArray y,
                          INDArray z,
                          int... dimensions)

Broadcast greater than op. See: BroadcastGreaterThan

gte

public static INDArray gte(INDArray x,
                           INDArray y,
                           INDArray z,
                           int... dimensions)

Broadcast greater than or equal to op. See: BroadcastGreaterThanOrEqual

lt

public static INDArray lt(INDArray x,
                          INDArray y,
                          INDArray z,
                          int... dimensions)

Broadcast less than op. See: BroadcastLessThan

lte

public static INDArray lte(INDArray x,
                           INDArray y,
                           INDArray z,
                           int... dimensions)

Broadcast less than or equal to op. See: BroadcastLessThanOrEqual

mul

public static INDArray mul(INDArray x,
                           INDArray y,
                           INDArray z,
                           int... dimensions)

Broadcast element-wise multiply op. See: BroadcastMulOp

neq

public static INDArray neq(INDArray x,
                           INDArray y,
                           INDArray z,
                           int... dimensions)

Broadcast not equal to op. See: BroadcastNotEqual

rdiv

public static INDArray rdiv(INDArray x,
                            INDArray y,
                            INDArray z,
                            int... dimensions)

Broadcast reverse division op. See: BroadcastRDivOp

rsub

public static INDArray rsub(INDArray x,
                            INDArray y,
                            INDArray z,
                            int... dimensions)

Broadcast reverse subtraction op. See: BroadcastRSubOp

sub

public static INDArray sub(INDArray x,
                           INDArray y,
                           INDArray z,
                           int... dimensions)

Broadcast subtraction op. See: BroadcastSubOp

max

public static INDArray max(INDArray x,
                           INDArray y,
                           INDArray z,
                           int... dimensions)

Broadcast max op. See: BroadcastMax

min

public static INDArray min(INDArray x,
                           INDArray y,
                           INDArray z,
                           int... dimensions)

Broadcast min op. See: BroadcastMin

amax

public static INDArray amax(INDArray x,
                            INDArray y,
                            INDArray z,
                            int... dimensions)

Broadcast absolute max op. See: BroadcastAMax

amin

public static INDArray amin(INDArray x,
                            INDArray y,
                            INDArray z,
                            int... dimensions)

Broadcast absolute min op. See: BroadcastAMax

validateShapesNoDimCase

public static void validateShapesNoDimCase(INDArray x,
                                           INDArray y,
                                           INDArray z)

validateBroadcastDims

public static void validateBroadcastDims(INDArray x,
                                         INDArray y,
                                         INDArray z,
                                         int... dimensions)

Validate the broadcast dimensions for manual broadcast ops such as BroadcastMulOp. Here, the dimensions are those that the arrays match on WRT X. For example, mul([a,b,c], [a,c], 0,2)