ai.djl.mxnet.engine

Class MxNDArray16

java.lang.Object

ai.djl.util.NativeResource<com.sun.jna.Pointer>

ai.djl.mxnet.engine.MxNDArray

ai.djl.mxnet.engine.MxNDArray16

All Implemented Interfaces:

ai.djl.ndarray.LazyNDArray, ai.djl.ndarray.NDArray, java.lang.AutoCloseable

public class MxNDArray16
extends MxNDArray

MXNet 1.6.0 MxNDArray implementation.

Field Summary

Fields inherited from class ai.djl.mxnet.engine.MxNDArray

manager

Fields inherited from class ai.djl.util.NativeResource

handle

Method Summary

Modifier and Type	Method and Description
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	broadcast(ai.djl.ndarray.types.Shape shape)
ai.djl.ndarray.NDArray	expandDims(int axis)
ai.djl.ndarray.NDArray	isNaN()
ai.djl.ndarray.NDArray	matMul(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray	onesLike()
ai.djl.ndarray.NDArray	sort()
ai.djl.ndarray.NDArray	sort(int axis)
ai.djl.ndarray.NDArray	zerosLike()

Methods inherited from class ai.djl.mxnet.engine.MxNDArray

abs, acos, acosh, add, add, addi, addi, asin, asinh, atan, atanh, attach, attachGradient, attachGradient, backward, booleanMask, cbrt, ceil, clip, close, contentEquals, contentEquals, copyTo, cos, cosh, cumSum, cumSum, detach, div, div, divi, divi, dot, eq, eq, equals, erfinv, exp, flatten, flip, floor, getDataType, getDevice, getGradient, getManager, getName, getNDArrayInternal, getShape, getSparseFormat, gt, gt, gte, gte, hasGradient, hashCode, isInfinite, log, log10, log2, logicalAnd, logicalNot, logicalOr, logicalXor, logSoftmax, lt, lt, lte, lte, max, max, max, maximum, maximum, mean, mean, median, median, min, min, minimum, minimum, mod, mod, modi, modi, mul, mul, muli, muli, neg, negi, neq, neq, nonzero, percentile, percentile, pow, pow, powi, powi, prod, prod, repeat, repeat, repeat, repeat, reshape, round, sequenceMask, sequenceMask, set, setName, sign, signi, sin, sinh, softmax, split, sqrt, square, squeeze, squeeze, sub, sub, subi, subi, sum, sum, swapAxes, tan, tanh, tile, tile, tile, tile, toByteBuffer, toDegrees, toDense, toDevice, toRadians, toSparse, toString, toType, trace, transpose, transpose, trunc, waitAll, waitToRead, waitToWrite

Methods inherited from class ai.djl.util.NativeResource

getHandle, getUid, isReleased

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Methods inherited from interface ai.djl.ndarray.NDArray

all, 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, trace

Method Detail

zerosLike

public ai.djl.ndarray.NDArray zerosLike()

Specified by:

zerosLike in interface ai.djl.ndarray.NDArray

Overrides:

zerosLike in class MxNDArray

onesLike

public ai.djl.ndarray.NDArray onesLike()

Specified by:

onesLike in interface ai.djl.ndarray.NDArray

Overrides:

onesLike in class MxNDArray

argSort

public ai.djl.ndarray.NDArray argSort(int axis,
                                      boolean ascending)

Specified by:

argSort in interface ai.djl.ndarray.NDArray

Overrides:

argSort in class MxNDArray

sort

public ai.djl.ndarray.NDArray sort(int axis)

Specified by:

sort in interface ai.djl.ndarray.NDArray

Overrides:

sort in class MxNDArray

sort

public ai.djl.ndarray.NDArray sort()

Specified by:

sort in interface ai.djl.ndarray.NDArray

Overrides:

sort in class MxNDArray

argMax

public ai.djl.ndarray.NDArray argMax()

Specified by:

argMax in interface ai.djl.ndarray.NDArray

Overrides:

argMax in class MxNDArray

argMax

public ai.djl.ndarray.NDArray argMax(int axis)

Specified by:

argMax in interface ai.djl.ndarray.NDArray

Overrides:

argMax in class MxNDArray

argMin

public ai.djl.ndarray.NDArray argMin()

Specified by:

argMin in interface ai.djl.ndarray.NDArray

Overrides:

argMin in class MxNDArray

argMin

public ai.djl.ndarray.NDArray argMin(int axis)

Specified by:

argMin in interface ai.djl.ndarray.NDArray

Overrides:

argMin in class MxNDArray

expandDims

public ai.djl.ndarray.NDArray expandDims(int axis)

Specified by:

expandDims in interface ai.djl.ndarray.NDArray

Overrides:

expandDims in class MxNDArray

matMul

public ai.djl.ndarray.NDArray matMul(ai.djl.ndarray.NDArray other)

Specified by:

matMul in interface ai.djl.ndarray.NDArray

Overrides:

matMul in class MxNDArray

isNaN

public ai.djl.ndarray.NDArray isNaN()

Specified by:

isNaN in interface ai.djl.ndarray.NDArray

Overrides:

isNaN in class MxNDArray

broadcast

public ai.djl.ndarray.NDArray broadcast(ai.djl.ndarray.types.Shape shape)

Specified by:

broadcast in interface ai.djl.ndarray.NDArray

Overrides:

broadcast in class MxNDArray