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 TypeMethodDescriptionvoid
waitAll()
Runs all NDArrays and sleeps until their values are fully computed.void
Runs the current NDArray and sleeps until the value is ready to read.void
Runs the current NDArray and sleeps until the value is ready to write.Methods inherited from interface ai.djl.ndarray.BytesSupplier
getAsBytes, getAsObject, getAsString
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, 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, zerosLike
Methods 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.
-