ai.djl.ndarray

Interface LazyNDArray

All Superinterfaces:

java.lang.AutoCloseable, NDArray

public interface LazyNDArray
extends NDArray

An NDArray that waits to compute values until they are needed.

Method Summary

Modifier and Type	Method and Description
void	waitAll() Runs all NDArrays and sleeps until their values are fully computed.
void	waitToRead() Runs the current NDArray and sleeps until the value is ready to read.
void	waitToWrite() Runs the current NDArray and sleeps until the value is ready to write.

Methods inherited from interface ai.djl.ndarray.NDArray

abs, acos, acosh, add, add, addi, addi, all, allClose, allClose, any, argMax, argMax, argMin, argMin, argSort, argSort, argSort, asin, asinh, atan, atanh, attach, attachGradient, attachGradient, booleanMask, booleanMask, broadcast, broadcast, cbrt, ceil, clip, close, concat, concat, contentEquals, contentEquals, copyTo, cos, cosh, countNonzero, countNonzero, createMask, createMask, cumSum, cumSum, decode, detach, div, div, divi, divi, dot, duplicate, encode, eq, eq, exp, expandDims, flatten, floor, get, get, get, get, getBoolean, getByte, getDataType, getDevice, getDouble, getFloat, getGradient, getInt, getLong, getManager, getName, getNDArrayInternal, getScalar, getShape, getSparseFormat, getUid, getUint8, gt, gt, gte, gte, isEmpty, isInfinite, isNaN, isScalar, isSparse, like, log, log10, log2, logicalAnd, logicalNot, logicalOr, logicalXor, logSoftmax, logSoftmax, logSoftmax, logSoftmax, lt, lt, lte, lte, matMul, max, max, max, maximum, maximum, mean, mean, mean, median, median, min, min, min, minimum, minimum, mod, mod, modi, modi, mul, mul, muli, muli, neg, negi, neq, neq, none, nonzero, onesLike, percentile, percentile, pow, pow, powi, powi, prod, prod, prod, repeat, repeat, repeat, repeat, reshape, reshape, reshapeLike, round, sequenceMask, sequenceMask, set, set, set, set, set, set, set, set, set, setName, setScalar, shapeEquals, sin, sinh, size, size, softmax, softmax, softmax, softmax, sort, sort, split, split, split, split, sqrt, square, squeeze, squeeze, squeeze, stack, stack, sub, sub, subi, subi, sum, sum, sum, swapAxes, tan, tanh, tile, tile, tile, tile, toArray, toBooleanArray, toByteArray, toByteBuffer, toDebugString, toDegrees, toDense, toDevice, toDoubleArray, toFloatArray, toIntArray, toLongArray, toRadians, toSparse, toType, toUint8Array, trace, trace, trace, transpose, transpose, trunc, zerosLike

Method Detail

waitToRead

void waitToRead()

Runs the current NDArray and sleeps until the value is ready to read.

waitToWrite

void waitToWrite()

Runs the current NDArray and sleeps until the value is ready to write.

waitAll

void waitAll()

Runs all NDArrays and sleeps until their values are fully computed.