Package ai.djl.ndarray
Interface LazyNDArray
- All Superinterfaces:
AutoCloseable,BytesSupplier,NDArray,NDResource
An
NDArray that waits to compute values until they are needed.-
Method Summary
Modifier and TypeMethodDescriptionvoidwaitAll()Runs all NDArrays and sleeps until their values are fully computed.voidRuns the current NDArray and sleeps until the value is ready to read.voidRuns the current NDArray and sleeps until the value is ready to write.Methods inherited from interface ai.djl.ndarray.BytesSupplier
getAsBytes, getAsObject, getAsStringMethods 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, atan2, atanh, batchDot, batchMatMul, booleanMask, booleanMask, broadcast, broadcast, cbrt, ceil, clip, close, complex, concat, concat, conj, contentEquals, contentEquals, copyTo, cos, cosh, countNonzero, countNonzero, cumProd, cumProd, cumSum, cumSum, div, div, divi, divi, dot, duplicate, encode, eq, eq, erf, erfinv, exp, expandDims, fft, fft, fft2, fft2, 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, ifft, ifft, ifft2, ifft2, intern, inverse, irfft, irfft, 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, pad, percentile, percentile, pow, pow, powi, powi, prod, prod, prod, put, real, repeat, repeat, repeat, repeat, reshape, reshape, rfft, rfft, 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, toByteBuffer, toByteBuffer, 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, zerosLikeMethods inherited from interface ai.djl.ndarray.NDResource
attach, detach, getManager, returnResource, tempAttach
-
Method Details
-
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.
-