Package ai.djl.ndarray

Interface LazyNDArray

All Superinterfaces:

java.lang.AutoCloseable, BytesSupplier, NDArray, NDResource

public interface LazyNDArray
extends NDArray

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

Method Summary

Modifier and Type	Method	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.BytesSupplier

getAsBytes, getAsObject, getAsString, toByteBuffer

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, batchDot, batchMatMul, booleanMask, booleanMask, broadcast, broadcast, cbrt, ceil, clip, close, complex, concat, concat, contentEquals, contentEquals, copyTo, cos, cosh, countNonzero, countNonzero, cumProd, cumProd, cumSum, cumSum, div, div, divi, divi, dot, duplicate, encode, eq, eq, erfinv, exp, expandDims, fft, fft, flatten, flatten, flip, floor, gammaln, gather, gatherNd, get, get, get, get, get, get, getBoolean, getByte, getDataType, getDevice, getDouble, getFloat, getGradient, getInt, getLong, getName, getNDArrayInternal, getResourceNDArrays, getScalar, getShape, getSparseFormat, getUid, getUint8, gt, gt, gte, gte, hasGradient, intern, inverse, isEmpty, isInfinite, isNaN, isReleased, isScalar, isSparse, like, log, log10, log2, logicalAnd, logicalNot, logicalOr, logicalXor, 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, norm, norm, norm, norm, norm, normalize, normalize, normalize, oneHot, oneHot, oneHot, onesLike, percentile, percentile, pow, pow, powi, powi, prod, prod, prod, put, real, repeat, repeat, repeat, repeat, reshape, reshape, rotate90, round, scaleGradient, scatter, sequenceMask, sequenceMask, set, set, set, set, set, set, set, set, set, set, setName, setRequiresGradient, setScalar, shapeEquals, sign, signi, sin, sinh, size, size, softmax, sort, sort, split, split, split, split, sqrt, square, squeeze, squeeze, squeeze, stack, stack, stft, stft, stopGradient, sub, sub, subi, subi, sum, sum, sum, swapAxes, take, take, tan, tanh, tile, tile, tile, tile, toArray, toBooleanArray, toByteArray, toDebugString, toDebugString, toDebugString, toDegrees, toDense, toDevice, toDoubleArray, toFloatArray, toIntArray, toLongArray, topK, topK, toRadians, toShortArray, toSparse, toStringArray, toStringArray, toType, toUint8Array, toUnsignedIntArray, toUnsignedShortArray, trace, trace, trace, transpose, transpose, trunc, unique, unique, unique, xlogy, zerosLike

Methods inherited from interface ai.djl.ndarray.NDResource

attach, detach, getManager, returnResource, tempAttach

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.