public class Broadcast extends Object
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) |
public static INDArray add(INDArray x, INDArray y, INDArray z, int... dimensions)
BroadcastAddOp
public static INDArray copy(INDArray x, INDArray y, INDArray z, int... dimensions)
BroadcastCopyOp
public static INDArray div(INDArray x, INDArray y, INDArray z, int... dimensions)
BroadcastDivOp
public static INDArray eq(INDArray x, INDArray y, INDArray z, int... dimensions)
BroadcastEqualTo
public static INDArray gt(INDArray x, INDArray y, INDArray z, int... dimensions)
BroadcastGreaterThan
public static INDArray gte(INDArray x, INDArray y, INDArray z, int... dimensions)
BroadcastGreaterThanOrEqual
public static INDArray lt(INDArray x, INDArray y, INDArray z, int... dimensions)
BroadcastLessThan
public static INDArray lte(INDArray x, INDArray y, INDArray z, int... dimensions)
BroadcastLessThanOrEqual
public static INDArray mul(INDArray x, INDArray y, INDArray z, int... dimensions)
BroadcastMulOp
public static INDArray neq(INDArray x, INDArray y, INDArray z, int... dimensions)
BroadcastNotEqual
public static INDArray rdiv(INDArray x, INDArray y, INDArray z, int... dimensions)
BroadcastRDivOp
public static INDArray rsub(INDArray x, INDArray y, INDArray z, int... dimensions)
BroadcastRSubOp
public static INDArray sub(INDArray x, INDArray y, INDArray z, int... dimensions)
BroadcastSubOp
public static INDArray max(INDArray x, INDArray y, INDArray z, int... dimensions)
BroadcastMax
public static INDArray min(INDArray x, INDArray y, INDArray z, int... dimensions)
BroadcastMin
public static INDArray amax(INDArray x, INDArray y, INDArray z, int... dimensions)
BroadcastAMax
public static INDArray amin(INDArray x, INDArray y, INDArray z, int... dimensions)
BroadcastAMax
public static void validateShapesNoDimCase(INDArray x, INDArray y, INDArray z)
public static void validateBroadcastDims(INDArray x, INDArray y, INDArray z, int... dimensions)
BroadcastMulOp
.
Here, the dimensions are those that the arrays match on WRT X.
For example, mul([a,b,c], [a,c], 0,2)Copyright © 2021. All rights reserved.