public class MxNDArray extends NativeResource implements ai.djl.ndarray.LazyNDArray
MxNDArray is the MXNet implementation of NDArray.handle| Modifier and Type | Method and Description |
|---|---|
ai.djl.ndarray.NDArray |
abs() |
ai.djl.ndarray.NDArray |
acos() |
ai.djl.ndarray.NDArray |
acosh() |
ai.djl.ndarray.NDArray |
add(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
add(java.lang.Number n) |
ai.djl.ndarray.NDArray |
addi(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
addi(java.lang.Number n) |
ai.djl.ndarray.NDArray |
argMax() |
ai.djl.ndarray.NDArray |
argMax(int axis) |
ai.djl.ndarray.NDArray |
argMin() |
ai.djl.ndarray.NDArray |
argMin(int axis) |
ai.djl.ndarray.NDArray |
argSort(int axis,
boolean ascending) |
ai.djl.ndarray.NDArray |
asin() |
ai.djl.ndarray.NDArray |
asinh() |
ai.djl.ndarray.NDArray |
atan() |
ai.djl.ndarray.NDArray |
atanh() |
void |
attach(ai.djl.ndarray.NDManager manager) |
void |
attachGradient() |
void |
attachGradient(ai.djl.ndarray.types.SparseFormat sparseFormat) |
void |
backward(boolean retainGraph)
Computes the gradients of the NDArray w.r.t variables.
|
ai.djl.ndarray.NDArray |
booleanMask(ai.djl.ndarray.NDArray index,
int axis) |
ai.djl.ndarray.NDArray |
broadcast(ai.djl.ndarray.types.Shape shape) |
ai.djl.ndarray.NDArray |
cbrt() |
ai.djl.ndarray.NDArray |
ceil() |
ai.djl.ndarray.NDArray |
clip(java.lang.Number min,
java.lang.Number max) |
void |
close() |
boolean |
contentEquals(ai.djl.ndarray.NDArray other) |
boolean |
contentEquals(java.lang.Number number) |
void |
copyTo(ai.djl.ndarray.NDArray ndArray) |
ai.djl.ndarray.NDArray |
cos() |
ai.djl.ndarray.NDArray |
cosh() |
ai.djl.ndarray.NDArray |
cumSum() |
ai.djl.ndarray.NDArray |
cumSum(int axis) |
void |
detach() |
ai.djl.ndarray.NDArray |
div(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
div(java.lang.Number n) |
ai.djl.ndarray.NDArray |
divi(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
divi(java.lang.Number n) |
ai.djl.ndarray.NDArray |
dot(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
eq(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
eq(java.lang.Number n) |
boolean |
equals(java.lang.Object obj) |
ai.djl.ndarray.NDArray |
exp() |
ai.djl.ndarray.NDArray |
expandDims(int axis) |
ai.djl.ndarray.NDArray |
flatten() |
ai.djl.ndarray.NDArray |
floor() |
ai.djl.ndarray.types.DataType |
getDataType() |
ai.djl.Device |
getDevice() |
ai.djl.ndarray.NDArray |
getGradient() |
MxNDManager |
getManager() |
java.lang.String |
getName() |
ai.djl.ndarray.internal.NDArrayEx |
getNDArrayInternal() |
ai.djl.ndarray.types.Shape |
getShape() |
ai.djl.ndarray.types.SparseFormat |
getSparseFormat() |
ai.djl.ndarray.NDArray |
gt(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
gt(java.lang.Number other) |
ai.djl.ndarray.NDArray |
gte(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
gte(java.lang.Number other) |
boolean |
hasGradient() |
int |
hashCode() |
ai.djl.ndarray.NDArray |
isInfinite() |
ai.djl.ndarray.NDArray |
isNaN() |
ai.djl.ndarray.NDArray |
log() |
ai.djl.ndarray.NDArray |
log10() |
ai.djl.ndarray.NDArray |
log2() |
ai.djl.ndarray.NDArray |
logicalAnd(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
logicalNot() |
ai.djl.ndarray.NDArray |
logicalOr(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
logicalXor(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
logSoftmax(int axis) |
ai.djl.ndarray.NDArray |
lt(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
lt(java.lang.Number other) |
ai.djl.ndarray.NDArray |
lte(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
lte(java.lang.Number other) |
ai.djl.ndarray.NDArray |
matMul(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
max() |
ai.djl.ndarray.NDArray |
max(int[] axes) |
ai.djl.ndarray.NDArray |
max(int[] axes,
boolean keepDims) |
ai.djl.ndarray.NDArray |
maximum(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
maximum(java.lang.Number n) |
ai.djl.ndarray.NDArray |
mean() |
ai.djl.ndarray.NDArray |
mean(int[] axes,
boolean keepDims) |
ai.djl.ndarray.NDArray |
median() |
ai.djl.ndarray.NDArray |
median(int[] axes) |
ai.djl.ndarray.NDArray |
min() |
ai.djl.ndarray.NDArray |
min(int[] axes,
boolean keepDims) |
ai.djl.ndarray.NDArray |
minimum(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
minimum(java.lang.Number n) |
ai.djl.ndarray.NDArray |
mod(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
mod(java.lang.Number n) |
ai.djl.ndarray.NDArray |
modi(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
modi(java.lang.Number n) |
ai.djl.ndarray.NDArray |
mul(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
mul(java.lang.Number n) |
ai.djl.ndarray.NDArray |
muli(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
muli(java.lang.Number n) |
ai.djl.ndarray.NDArray |
neg() |
ai.djl.ndarray.NDArray |
negi() |
ai.djl.ndarray.NDArray |
neq(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
neq(java.lang.Number other) |
ai.djl.ndarray.NDArray |
nonzero() |
ai.djl.ndarray.NDArray |
onesLike() |
ai.djl.ndarray.NDArray |
percentile(java.lang.Number percentile) |
ai.djl.ndarray.NDArray |
percentile(java.lang.Number percentile,
int[] dimension) |
ai.djl.ndarray.NDArray |
pow(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
pow(java.lang.Number n) |
ai.djl.ndarray.NDArray |
powi(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
powi(java.lang.Number n) |
ai.djl.ndarray.NDArray |
prod() |
ai.djl.ndarray.NDArray |
prod(int[] axes,
boolean keepDims) |
ai.djl.ndarray.NDArray |
repeat(int axis,
long repeats) |
ai.djl.ndarray.NDArray |
repeat(long repeats) |
ai.djl.ndarray.NDArray |
repeat(long[] repeats) |
ai.djl.ndarray.NDArray |
repeat(ai.djl.ndarray.types.Shape desiredShape) |
ai.djl.ndarray.NDArray |
reshape(ai.djl.ndarray.types.Shape shape) |
ai.djl.ndarray.NDArray |
round() |
ai.djl.ndarray.NDArray |
sequenceMask(ai.djl.ndarray.NDArray sequenceLength) |
ai.djl.ndarray.NDArray |
sequenceMask(ai.djl.ndarray.NDArray sequenceLength,
float value) |
void |
set(java.nio.Buffer data) |
void |
setName(java.lang.String name) |
void |
setShouldFree(boolean shouldFree)
Sets whether to free the MxNDArray when it is closed (internal).
|
ai.djl.ndarray.NDArray |
sin() |
ai.djl.ndarray.NDArray |
sinh() |
ai.djl.ndarray.NDArray |
softmax(int axis) |
ai.djl.ndarray.NDArray |
sort() |
ai.djl.ndarray.NDArray |
sort(int axis) |
ai.djl.ndarray.NDList |
split(long[] indices,
int axis) |
ai.djl.ndarray.NDArray |
sqrt() |
ai.djl.ndarray.NDArray |
square() |
ai.djl.ndarray.NDArray |
squeeze() |
ai.djl.ndarray.NDArray |
squeeze(int[] axes) |
ai.djl.ndarray.NDArray |
sub(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
sub(java.lang.Number n) |
ai.djl.ndarray.NDArray |
subi(ai.djl.ndarray.NDArray other) |
ai.djl.ndarray.NDArray |
subi(java.lang.Number n) |
ai.djl.ndarray.NDArray |
sum() |
ai.djl.ndarray.NDArray |
sum(int[] axes,
boolean keepDims) |
ai.djl.ndarray.NDArray |
swapAxes(int axis1,
int axis2) |
ai.djl.ndarray.NDArray |
tan() |
ai.djl.ndarray.NDArray |
tanh() |
ai.djl.ndarray.NDArray |
tile(int axis,
long repeats) |
ai.djl.ndarray.NDArray |
tile(long repeats) |
ai.djl.ndarray.NDArray |
tile(long[] repeats) |
ai.djl.ndarray.NDArray |
tile(ai.djl.ndarray.types.Shape desiredShape) |
java.nio.ByteBuffer |
toByteBuffer() |
ai.djl.ndarray.NDArray |
toDegrees() |
ai.djl.ndarray.NDArray |
toDense() |
ai.djl.ndarray.NDArray |
toDevice(ai.djl.Device device,
boolean copy) |
ai.djl.ndarray.NDArray |
toRadians() |
ai.djl.ndarray.NDArray |
toSparse(ai.djl.ndarray.types.SparseFormat fmt) |
java.lang.String |
toString() |
ai.djl.ndarray.NDArray |
toType(ai.djl.ndarray.types.DataType dataType,
boolean copy) |
ai.djl.ndarray.NDArray |
trace(int offset,
int axis1,
int axis2) |
ai.djl.ndarray.NDArray |
transpose() |
ai.djl.ndarray.NDArray |
transpose(int... dimensions) |
ai.djl.ndarray.NDArray |
trunc() |
void |
waitAll() |
void |
waitToRead() |
void |
waitToWrite() |
ai.djl.ndarray.NDArray |
zerosLike() |
finalize, getHandle, getUid, isReleasedall, allClose, allClose, any, argSort, argSort, booleanMask, broadcast, concat, concat, countNonzero, countNonzero, decode, duplicate, encode, get, get, get, get, getBoolean, getByte, getDouble, getFloat, getInt, getLong, getScalar, getUid, getUint8, isEmpty, isScalar, isSparse, like, mean, min, none, prod, reshape, set, set, set, set, set, set, set, set, setScalar, shapeEquals, size, size, split, split, split, squeeze, stack, stack, sum, toArray, toBooleanArray, toByteArray, toDebugString, toDoubleArray, toFloatArray, toIntArray, toLongArray, toUint8Array, trace, tracepublic MxNDManager getManager()
getManager in interface ai.djl.ndarray.NDArraypublic java.lang.String getName()
getName in interface ai.djl.ndarray.NDArraypublic void setName(java.lang.String name)
setName in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.types.DataType getDataType()
getDataType in interface ai.djl.ndarray.NDArraypublic ai.djl.Device getDevice()
getDevice in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.types.Shape getShape()
getShape in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.types.SparseFormat getSparseFormat()
getSparseFormat in interface ai.djl.ndarray.NDArraypublic void attach(ai.djl.ndarray.NDManager manager)
attach in interface ai.djl.ndarray.NDArraypublic void detach()
detach in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray toDevice(ai.djl.Device device,
boolean copy)
toDevice in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray toType(ai.djl.ndarray.types.DataType dataType,
boolean copy)
toType in interface ai.djl.ndarray.NDArraypublic void setShouldFree(boolean shouldFree)
It should not be freed in cases such as MxParameterServer optimizer callback where the NDArray is merely intended to be read, not freed. Otherwise, leave it as the deafult (should free).
shouldFree - true if the MxNDArray should be freed on closepublic void backward(boolean retainGraph)
retainGraph - whether to retain the computation graph for another backward pass on the
same graph. By default, the computation history is cleared.public void attachGradient()
attachGradient in interface ai.djl.ndarray.NDArraypublic void attachGradient(ai.djl.ndarray.types.SparseFormat sparseFormat)
attachGradient in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray getGradient()
getGradient in interface ai.djl.ndarray.NDArraypublic boolean hasGradient()
hasGradient in interface ai.djl.ndarray.NDArraypublic java.nio.ByteBuffer toByteBuffer()
toByteBuffer in interface ai.djl.ndarray.NDArraypublic void set(java.nio.Buffer data)
set in interface ai.djl.ndarray.NDArraypublic void copyTo(ai.djl.ndarray.NDArray ndArray)
copyTo in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray booleanMask(ai.djl.ndarray.NDArray index,
int axis)
booleanMask in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray sequenceMask(ai.djl.ndarray.NDArray sequenceLength,
float value)
sequenceMask in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray sequenceMask(ai.djl.ndarray.NDArray sequenceLength)
sequenceMask in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray zerosLike()
zerosLike in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray onesLike()
onesLike in interface ai.djl.ndarray.NDArraypublic boolean contentEquals(java.lang.Number number)
contentEquals in interface ai.djl.ndarray.NDArraypublic boolean contentEquals(ai.djl.ndarray.NDArray other)
contentEquals in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray eq(java.lang.Number n)
eq in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray eq(ai.djl.ndarray.NDArray other)
eq in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray neq(java.lang.Number other)
neq in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray neq(ai.djl.ndarray.NDArray other)
neq in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray gt(java.lang.Number other)
gt in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray gt(ai.djl.ndarray.NDArray other)
gt in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray gte(java.lang.Number other)
gte in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray gte(ai.djl.ndarray.NDArray other)
gte in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray lt(java.lang.Number other)
lt in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray lt(ai.djl.ndarray.NDArray other)
lt in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray lte(java.lang.Number other)
lte in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray lte(ai.djl.ndarray.NDArray other)
lte in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray add(java.lang.Number n)
add in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray add(ai.djl.ndarray.NDArray other)
add in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray sub(java.lang.Number n)
sub in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray sub(ai.djl.ndarray.NDArray other)
sub in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray mul(java.lang.Number n)
mul in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray mul(ai.djl.ndarray.NDArray other)
mul in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray toSparse(ai.djl.ndarray.types.SparseFormat fmt)
toSparse in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray div(java.lang.Number n)
div in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray div(ai.djl.ndarray.NDArray other)
div in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray mod(java.lang.Number n)
mod in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray mod(ai.djl.ndarray.NDArray other)
mod in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray pow(java.lang.Number n)
pow in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray pow(ai.djl.ndarray.NDArray other)
pow in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray addi(java.lang.Number n)
addi in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray addi(ai.djl.ndarray.NDArray other)
addi in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray subi(java.lang.Number n)
subi in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray subi(ai.djl.ndarray.NDArray other)
subi in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray muli(java.lang.Number n)
muli in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray muli(ai.djl.ndarray.NDArray other)
muli in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray divi(java.lang.Number n)
divi in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray divi(ai.djl.ndarray.NDArray other)
divi in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray modi(java.lang.Number n)
modi in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray modi(ai.djl.ndarray.NDArray other)
modi in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray powi(java.lang.Number n)
powi in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray powi(ai.djl.ndarray.NDArray other)
powi in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray neg()
neg in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray negi()
negi in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray abs()
abs in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray square()
square in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray sqrt()
sqrt in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray cbrt()
cbrt in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray floor()
floor in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray ceil()
ceil in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray round()
round in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray trunc()
trunc in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray exp()
exp in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray log()
log in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray log10()
log10 in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray log2()
log2 in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray sin()
sin in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray cos()
cos in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray tan()
tan in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray asin()
asin in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray acos()
acos in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray atan()
atan in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray sinh()
sinh in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray cosh()
cosh in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray tanh()
tanh in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray asinh()
asinh in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray acosh()
acosh in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray atanh()
atanh in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray toDegrees()
toDegrees in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray toRadians()
toRadians in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray maximum(java.lang.Number n)
maximum in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray maximum(ai.djl.ndarray.NDArray other)
maximum in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray minimum(java.lang.Number n)
minimum in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray minimum(ai.djl.ndarray.NDArray other)
minimum in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray max()
max in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray max(int[] axes)
max in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray max(int[] axes,
boolean keepDims)
max in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray min()
min in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray min(int[] axes,
boolean keepDims)
min in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray sum()
sum in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray sum(int[] axes,
boolean keepDims)
sum in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray prod()
prod in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray prod(int[] axes,
boolean keepDims)
prod in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray mean()
mean in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray mean(int[] axes,
boolean keepDims)
mean in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray trace(int offset,
int axis1,
int axis2)
trace in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDList split(long[] indices,
int axis)
split in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray flatten()
flatten in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray reshape(ai.djl.ndarray.types.Shape shape)
reshape in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray expandDims(int axis)
expandDims in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray squeeze()
squeeze in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray squeeze(int[] axes)
squeeze in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray logicalAnd(ai.djl.ndarray.NDArray other)
logicalAnd in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray logicalOr(ai.djl.ndarray.NDArray other)
logicalOr in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray logicalXor(ai.djl.ndarray.NDArray other)
logicalXor in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray logicalNot()
logicalNot in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray argSort(int axis,
boolean ascending)
argSort in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray sort(int axis)
sort in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray sort()
sort in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray softmax(int axis)
softmax in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray logSoftmax(int axis)
logSoftmax in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray cumSum()
cumSum in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray cumSum(int axis)
cumSum in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray isInfinite()
isInfinite in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray isNaN()
isNaN in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray toDense()
toDense in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray tile(long repeats)
tile in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray tile(int axis,
long repeats)
tile in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray tile(long[] repeats)
tile in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray tile(ai.djl.ndarray.types.Shape desiredShape)
tile in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray repeat(long repeats)
repeat in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray repeat(int axis,
long repeats)
repeat in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray repeat(long[] repeats)
repeat in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray repeat(ai.djl.ndarray.types.Shape desiredShape)
repeat in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray dot(ai.djl.ndarray.NDArray other)
dot in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray matMul(ai.djl.ndarray.NDArray other)
matMul in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray clip(java.lang.Number min,
java.lang.Number max)
clip in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray swapAxes(int axis1,
int axis2)
swapAxes in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray transpose()
transpose in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray transpose(int... dimensions)
transpose in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray broadcast(ai.djl.ndarray.types.Shape shape)
broadcast in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray argMax()
argMax in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray argMax(int axis)
argMax in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray argMin()
argMin in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray argMin(int axis)
argMin in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray percentile(java.lang.Number percentile)
percentile in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray percentile(java.lang.Number percentile,
int[] dimension)
percentile in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray median()
median in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray median(int[] axes)
median in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.NDArray nonzero()
nonzero in interface ai.djl.ndarray.NDArraypublic ai.djl.ndarray.internal.NDArrayEx getNDArrayInternal()
getNDArrayInternal in interface ai.djl.ndarray.NDArraypublic void waitToRead()
waitToRead in interface ai.djl.ndarray.LazyNDArraypublic void waitToWrite()
waitToWrite in interface ai.djl.ndarray.LazyNDArraypublic void waitAll()
waitAll in interface ai.djl.ndarray.LazyNDArraypublic boolean equals(java.lang.Object obj)
equals in class java.lang.Objectpublic int hashCode()
hashCode in class java.lang.Objectpublic java.lang.String toString()
toString in class java.lang.Objectpublic void close()
close in interface ai.djl.ndarray.NDArrayclose in interface java.lang.AutoCloseableclose in class NativeResource