Package ai.djl.pytorch.engine
Class PtNDArray
- java.lang.Object
-
- ai.djl.util.NativeResource<java.lang.Long>
-
- ai.djl.pytorch.engine.PtNDArray
-
- All Implemented Interfaces:
ai.djl.ndarray.BytesSupplier,ai.djl.ndarray.NDArray,ai.djl.ndarray.NDResource,java.lang.AutoCloseable
public class PtNDArray extends ai.djl.util.NativeResource<java.lang.Long> implements ai.djl.ndarray.NDArrayPtNDArrayis the PyTorch implementation ofNDArray.
-
-
Constructor Summary
Constructors Constructor Description PtNDArray(PtNDManager manager, long handle)Constructs a PyTorchNDArrayfrom a native handle (internal.PtNDArray(PtNDManager manager, long handle, java.nio.ByteBuffer data)Constructs a PyTorchNDArrayfrom a native handle (internal.PtNDArray(PtNDManager manager, java.lang.String[] strs, ai.djl.ndarray.types.Shape shape)Constructs a PyTorchNDArrayto hold string array with a dummy native handle (internal.
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description PtNDArrayabs()PtNDArrayacos()PtNDArrayacosh()PtNDArrayadd(ai.djl.ndarray.NDArray other)PtNDArrayadd(java.lang.Number n)PtNDArrayaddi(ai.djl.ndarray.NDArray other)PtNDArrayaddi(java.lang.Number n)PtNDArrayall()PtNDArrayany()PtNDArrayargMax()PtNDArrayargMax(int axis)PtNDArrayargMin()PtNDArrayargMin(int axis)PtNDArrayargSort(int axis, boolean ascending)PtNDArrayasin()PtNDArrayasinh()PtNDArrayatan()PtNDArrayatanh()voidattach(ai.djl.ndarray.NDManager manager)ai.djl.ndarray.NDArraybatchDot(ai.djl.ndarray.NDArray other)ai.djl.ndarray.NDArraybatchMatMul(ai.djl.ndarray.NDArray other)PtNDArraybooleanMask(ai.djl.ndarray.NDArray index, int axis)PtNDArraybroadcast(ai.djl.ndarray.types.Shape shape)PtNDArraycbrt()PtNDArrayceil()PtNDArrayclip(java.lang.Number min, java.lang.Number max)voidclose()ai.djl.ndarray.NDArraycomplex()booleancontentEquals(ai.djl.ndarray.NDArray other)booleancontentEquals(java.lang.Number number)PtNDArraycos()PtNDArraycosh()ai.djl.ndarray.NDArraycumProd(int axis)ai.djl.ndarray.NDArraycumProd(int axis, ai.djl.ndarray.types.DataType dataType)PtNDArraycumSum()PtNDArraycumSum(int axis)voiddetach()PtNDArraydiv(ai.djl.ndarray.NDArray other)PtNDArraydiv(java.lang.Number n)PtNDArraydivi(ai.djl.ndarray.NDArray other)PtNDArraydivi(java.lang.Number n)PtNDArraydot(ai.djl.ndarray.NDArray other)ai.djl.ndarray.NDArrayduplicate()PtNDArrayeq(ai.djl.ndarray.NDArray other)PtNDArrayeq(java.lang.Number n)booleanequals(java.lang.Object obj)PtNDArrayerfinv()PtNDArrayexp()PtNDArrayexpandDims(int axis)ai.djl.ndarray.NDArrayfft(long length, long axis)PtNDArrayflatten()ai.djl.ndarray.NDArrayflatten(int startDim, int endDim)ai.djl.ndarray.NDArrayflip(int... axes)PtNDArrayfloor()ai.djl.ndarray.NDArraygammaln()ai.djl.ndarray.NDArraygather(ai.djl.ndarray.NDArray index, int axis)ai.djl.ndarray.NDArraygatherNd(ai.djl.ndarray.NDArray index)ai.djl.ndarray.NDArrayget(ai.djl.ndarray.NDManager manager, long... indices)ai.djl.ndarray.types.DataTypegetDataType()ai.djl.DevicegetDevice()PtNDArraygetGradient()PtNDManagergetManager()java.lang.StringgetName()PtNDArrayExgetNDArrayInternal()ai.djl.ndarray.types.ShapegetShape()ai.djl.ndarray.types.SparseFormatgetSparseFormat()PtNDArraygt(ai.djl.ndarray.NDArray other)PtNDArraygt(java.lang.Number n)PtNDArraygte(ai.djl.ndarray.NDArray other)PtNDArraygte(java.lang.Number n)booleanhasGradient()inthashCode()voidintern(ai.djl.ndarray.NDArray replaced)PtNDArrayinverse()PtNDArrayisInfinite()PtNDArrayisNaN()PtNDArraylog()PtNDArraylog10()PtNDArraylog2()PtNDArraylogicalAnd(ai.djl.ndarray.NDArray other)PtNDArraylogicalNot()PtNDArraylogicalOr(ai.djl.ndarray.NDArray other)PtNDArraylogicalXor(ai.djl.ndarray.NDArray other)PtNDArraylogSoftmax(int axis)PtNDArraylt(ai.djl.ndarray.NDArray other)PtNDArraylt(java.lang.Number n)PtNDArraylte(ai.djl.ndarray.NDArray other)PtNDArraylte(java.lang.Number n)ai.djl.ndarray.NDArraymatMul(ai.djl.ndarray.NDArray other)PtNDArraymax()PtNDArraymax(int[] axes, boolean keepDims)PtNDArraymaximum(ai.djl.ndarray.NDArray other)PtNDArraymaximum(java.lang.Number n)PtNDArraymean()PtNDArraymean(int[] axes, boolean keepDims)PtNDArraymedian()PtNDArraymedian(int[] axes)PtNDArraymin()PtNDArraymin(int[] axes, boolean keepDims)PtNDArrayminimum(ai.djl.ndarray.NDArray other)PtNDArrayminimum(java.lang.Number n)PtNDArraymod(ai.djl.ndarray.NDArray other)PtNDArraymod(java.lang.Number n)PtNDArraymodi(ai.djl.ndarray.NDArray other)PtNDArraymodi(java.lang.Number n)PtNDArraymul(ai.djl.ndarray.NDArray other)PtNDArraymul(java.lang.Number n)PtNDArraymuli(ai.djl.ndarray.NDArray other)PtNDArraymuli(java.lang.Number n)PtNDArrayneg()PtNDArraynegi()PtNDArrayneq(ai.djl.ndarray.NDArray other)PtNDArrayneq(java.lang.Number n)PtNDArraynone()PtNDArraynonzero()ai.djl.ndarray.NDArraynorm(boolean keepDims)ai.djl.ndarray.NDArraynorm(int order, int[] axes, boolean keepDims)PtNDArraynormalize(double p, long dim, double eps)ai.djl.ndarray.NDArrayoneHot(int depth)ai.djl.ndarray.NDArrayoneHot(int depth, float onValue, float offValue, ai.djl.ndarray.types.DataType dataType)ai.djl.ndarray.NDArrayoneHot(int depth, ai.djl.ndarray.types.DataType dataType)PtNDArraypercentile(java.lang.Number percentile)PtNDArraypercentile(java.lang.Number percentile, int[] axes)PtNDArraypow(ai.djl.ndarray.NDArray other)PtNDArraypow(java.lang.Number n)PtNDArraypowi(ai.djl.ndarray.NDArray other)PtNDArraypowi(java.lang.Number n)PtNDArrayprod()PtNDArrayprod(int[] axes, boolean keepDims)ai.djl.ndarray.NDArrayput(ai.djl.ndarray.NDArray index, ai.djl.ndarray.NDArray value)ai.djl.ndarray.NDArrayreal()PtNDArrayrepeat(int axis, long repeats)PtNDArrayrepeat(long repeats)PtNDArrayrepeat(long[] repeats)PtNDArrayrepeat(ai.djl.ndarray.types.Shape desiredShape)PtNDArrayreshape(ai.djl.ndarray.types.Shape shape)voidreturnResource(ai.djl.ndarray.NDManager manager)PtNDArrayrotate90(int times, int[] axes)PtNDArrayround()ai.djl.ndarray.NDArrayscatter(ai.djl.ndarray.NDArray index, ai.djl.ndarray.NDArray value, int axis)ai.djl.ndarray.NDArraysequenceMask(ai.djl.ndarray.NDArray sequenceLength)ai.djl.ndarray.NDArraysequenceMask(ai.djl.ndarray.NDArray sequenceLength, float value)voidset(java.nio.Buffer buffer)voidsetName(java.lang.String name)voidsetRequiresGradient(boolean requiresGrad)PtNDArraysign()PtNDArraysigni()PtNDArraysin()PtNDArraysinh()PtNDArraysoftmax(int axis)PtNDArraysort()PtNDArraysort(int axis)ai.djl.ndarray.NDListsplit(long[] indices, int axis)ai.djl.ndarray.NDListsplit(long sections, int axis)ai.djl.ndarray.NDArraysqrt()PtNDArraysquare()PtNDArraysqueeze()PtNDArraysqueeze(int axis)PtNDArraysqueeze(int[] axes)ai.djl.ndarray.NDArraystft(long nFft, long hopLength, boolean center, ai.djl.ndarray.NDArray window, boolean normalize, boolean returnComplex)ai.djl.ndarray.NDArraystopGradient()PtNDArraysub(ai.djl.ndarray.NDArray other)PtNDArraysub(java.lang.Number n)PtNDArraysubi(ai.djl.ndarray.NDArray other)PtNDArraysubi(java.lang.Number n)PtNDArraysum()PtNDArraysum(int[] axes, boolean keepDims)PtNDArrayswapAxes(int axis1, int axis2)ai.djl.ndarray.NDArraytake(ai.djl.ndarray.NDManager manager, ai.djl.ndarray.NDArray index)PtNDArraytan()PtNDArraytanh()voidtempAttach(ai.djl.ndarray.NDManager manager)PtNDArraytile(int axis, long repeats)PtNDArraytile(long repeats)PtNDArraytile(long[] repeats)PtNDArraytile(ai.djl.ndarray.types.Shape desiredShape)java.nio.ByteBuffertoByteBuffer()PtNDArraytoDegrees()PtNDArraytoDense()PtNDArraytoDevice(ai.djl.Device device, boolean copy)ai.djl.ndarray.NDListtopK(int k, int axis, boolean largest, boolean sorted)PtNDArraytoRadians()PtNDArraytoSparse(ai.djl.ndarray.types.SparseFormat fmt)java.lang.StringtoString()java.lang.String[]toStringArray(java.nio.charset.Charset charset)PtNDArraytoType(ai.djl.ndarray.types.DataType dataType, boolean copy)PtNDArraytrace(int offset, int axis1, int axis2)PtNDArraytranspose()PtNDArraytranspose(int... axes)PtNDArraytrunc()ai.djl.ndarray.NDListunique(java.lang.Integer dim, boolean sorted, boolean returnInverse, boolean returnCounts)ai.djl.ndarray.NDArrayxlogy(ai.djl.ndarray.NDArray other)-
Methods inherited from class java.lang.Object
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
-
Methods inherited from interface ai.djl.ndarray.NDArray
allClose, allClose, argSort, argSort, booleanMask, broadcast, concat, concat, copyTo, countNonzero, countNonzero, encode, fft, get, get, get, get, get, getBoolean, getByte, getDouble, getFloat, getInt, getLong, getResourceNDArrays, getScalar, getUid, getUint8, isEmpty, isReleased, isScalar, isSparse, like, max, mean, min, norm, norm, norm, normalize, normalize, onesLike, prod, reshape, scaleGradient, set, set, set, set, set, set, set, set, set, setScalar, shapeEquals, size, size, split, split, stack, stack, stft, sum, take, toArray, toBooleanArray, toByteArray, toDebugString, toDebugString, toDebugString, toDoubleArray, toFloatArray, toIntArray, toLongArray, topK, toShortArray, toStringArray, toUint8Array, toUnsignedIntArray, toUnsignedShortArray, trace, trace, unique, unique, zerosLike
-
-
-
-
Constructor Detail
-
PtNDArray
public PtNDArray(PtNDManager manager, long handle)
Constructs a PyTorchNDArrayfrom a native handle (internal. UseNDManagerinstead).- Parameters:
manager- the manager to attach the new array tohandle- the pointer to the native PyTorch memory
-
PtNDArray
public PtNDArray(PtNDManager manager, long handle, java.nio.ByteBuffer data)
Constructs a PyTorchNDArrayfrom a native handle (internal. UseNDManagerinstead) with the data that is hold on Java side.- Parameters:
manager- the manager to attach the new array tohandle- the pointer to the native PyTorch memorydata- the direct buffer of the data
-
PtNDArray
public PtNDArray(PtNDManager manager, java.lang.String[] strs, ai.djl.ndarray.types.Shape shape)
Constructs a PyTorchNDArrayto hold string array with a dummy native handle (internal. UseNDManagerinstead) with the data that is hold on Java side.- Parameters:
manager- the manager to attach the new array tostrs- the string arrayshape- theShapeof theNDArray
-
-
Method Detail
-
getManager
public PtNDManager getManager()
- Specified by:
getManagerin interfaceai.djl.ndarray.NDResource
-
getName
public java.lang.String getName()
- Specified by:
getNamein interfaceai.djl.ndarray.NDArray
-
setName
public void setName(java.lang.String name)
- Specified by:
setNamein interfaceai.djl.ndarray.NDArray
-
getDataType
public ai.djl.ndarray.types.DataType getDataType()
- Specified by:
getDataTypein interfaceai.djl.ndarray.NDArray
-
getDevice
public ai.djl.Device getDevice()
- Specified by:
getDevicein interfaceai.djl.ndarray.NDArray
-
getShape
public ai.djl.ndarray.types.Shape getShape()
- Specified by:
getShapein interfaceai.djl.ndarray.NDArray
-
getSparseFormat
public ai.djl.ndarray.types.SparseFormat getSparseFormat()
- Specified by:
getSparseFormatin interfaceai.djl.ndarray.NDArray
-
toDevice
public PtNDArray toDevice(ai.djl.Device device, boolean copy)
- Specified by:
toDevicein interfaceai.djl.ndarray.NDArray
-
toType
public PtNDArray toType(ai.djl.ndarray.types.DataType dataType, boolean copy)
- Specified by:
toTypein interfaceai.djl.ndarray.NDArray
-
setRequiresGradient
public void setRequiresGradient(boolean requiresGrad)
- Specified by:
setRequiresGradientin interfaceai.djl.ndarray.NDArray
-
getGradient
public PtNDArray getGradient()
- Specified by:
getGradientin interfaceai.djl.ndarray.NDArray
-
hasGradient
public boolean hasGradient()
- Specified by:
hasGradientin interfaceai.djl.ndarray.NDArray
-
stopGradient
public ai.djl.ndarray.NDArray stopGradient()
- Specified by:
stopGradientin interfaceai.djl.ndarray.NDArray
-
toByteBuffer
public java.nio.ByteBuffer toByteBuffer()
- Specified by:
toByteBufferin interfaceai.djl.ndarray.BytesSupplier
-
toStringArray
public java.lang.String[] toStringArray(java.nio.charset.Charset charset)
- Specified by:
toStringArrayin interfaceai.djl.ndarray.NDArray
-
set
public void set(java.nio.Buffer buffer)
- Specified by:
setin interfaceai.djl.ndarray.NDArray
-
get
public ai.djl.ndarray.NDArray get(ai.djl.ndarray.NDManager manager, long... indices)- Specified by:
getin interfaceai.djl.ndarray.NDArray
-
gather
public ai.djl.ndarray.NDArray gather(ai.djl.ndarray.NDArray index, int axis)- Specified by:
gatherin interfaceai.djl.ndarray.NDArray
-
gatherNd
public ai.djl.ndarray.NDArray gatherNd(ai.djl.ndarray.NDArray index)
- Specified by:
gatherNdin interfaceai.djl.ndarray.NDArray
-
take
public ai.djl.ndarray.NDArray take(ai.djl.ndarray.NDManager manager, ai.djl.ndarray.NDArray index)- Specified by:
takein interfaceai.djl.ndarray.NDArray
-
put
public ai.djl.ndarray.NDArray put(ai.djl.ndarray.NDArray index, ai.djl.ndarray.NDArray value)- Specified by:
putin interfaceai.djl.ndarray.NDArray
-
scatter
public ai.djl.ndarray.NDArray scatter(ai.djl.ndarray.NDArray index, ai.djl.ndarray.NDArray value, int axis)- Specified by:
scatterin interfaceai.djl.ndarray.NDArray
-
attach
public void attach(ai.djl.ndarray.NDManager manager)
- Specified by:
attachin interfaceai.djl.ndarray.NDResource
-
returnResource
public void returnResource(ai.djl.ndarray.NDManager manager)
- Specified by:
returnResourcein interfaceai.djl.ndarray.NDResource
-
tempAttach
public void tempAttach(ai.djl.ndarray.NDManager manager)
- Specified by:
tempAttachin interfaceai.djl.ndarray.NDResource
-
detach
public void detach()
- Specified by:
detachin interfaceai.djl.ndarray.NDResource
-
duplicate
public ai.djl.ndarray.NDArray duplicate()
- Specified by:
duplicatein interfaceai.djl.ndarray.NDArray
-
booleanMask
public PtNDArray booleanMask(ai.djl.ndarray.NDArray index, int axis)
- Specified by:
booleanMaskin interfaceai.djl.ndarray.NDArray
-
sequenceMask
public ai.djl.ndarray.NDArray sequenceMask(ai.djl.ndarray.NDArray sequenceLength, float value)- Specified by:
sequenceMaskin interfaceai.djl.ndarray.NDArray
-
sequenceMask
public ai.djl.ndarray.NDArray sequenceMask(ai.djl.ndarray.NDArray sequenceLength)
- Specified by:
sequenceMaskin interfaceai.djl.ndarray.NDArray
-
contentEquals
public boolean contentEquals(java.lang.Number number)
- Specified by:
contentEqualsin interfaceai.djl.ndarray.NDArray
-
contentEquals
public boolean contentEquals(ai.djl.ndarray.NDArray other)
- Specified by:
contentEqualsin interfaceai.djl.ndarray.NDArray
-
eq
public PtNDArray eq(java.lang.Number n)
- Specified by:
eqin interfaceai.djl.ndarray.NDArray
-
eq
public PtNDArray eq(ai.djl.ndarray.NDArray other)
- Specified by:
eqin interfaceai.djl.ndarray.NDArray
-
neq
public PtNDArray neq(java.lang.Number n)
- Specified by:
neqin interfaceai.djl.ndarray.NDArray
-
neq
public PtNDArray neq(ai.djl.ndarray.NDArray other)
- Specified by:
neqin interfaceai.djl.ndarray.NDArray
-
gt
public PtNDArray gt(java.lang.Number n)
- Specified by:
gtin interfaceai.djl.ndarray.NDArray
-
gt
public PtNDArray gt(ai.djl.ndarray.NDArray other)
- Specified by:
gtin interfaceai.djl.ndarray.NDArray
-
gte
public PtNDArray gte(java.lang.Number n)
- Specified by:
gtein interfaceai.djl.ndarray.NDArray
-
gte
public PtNDArray gte(ai.djl.ndarray.NDArray other)
- Specified by:
gtein interfaceai.djl.ndarray.NDArray
-
lt
public PtNDArray lt(java.lang.Number n)
- Specified by:
ltin interfaceai.djl.ndarray.NDArray
-
lt
public PtNDArray lt(ai.djl.ndarray.NDArray other)
- Specified by:
ltin interfaceai.djl.ndarray.NDArray
-
lte
public PtNDArray lte(java.lang.Number n)
- Specified by:
ltein interfaceai.djl.ndarray.NDArray
-
lte
public PtNDArray lte(ai.djl.ndarray.NDArray other)
- Specified by:
ltein interfaceai.djl.ndarray.NDArray
-
add
public PtNDArray add(java.lang.Number n)
- Specified by:
addin interfaceai.djl.ndarray.NDArray
-
add
public PtNDArray add(ai.djl.ndarray.NDArray other)
- Specified by:
addin interfaceai.djl.ndarray.NDArray
-
sub
public PtNDArray sub(java.lang.Number n)
- Specified by:
subin interfaceai.djl.ndarray.NDArray
-
sub
public PtNDArray sub(ai.djl.ndarray.NDArray other)
- Specified by:
subin interfaceai.djl.ndarray.NDArray
-
mul
public PtNDArray mul(java.lang.Number n)
- Specified by:
mulin interfaceai.djl.ndarray.NDArray
-
mul
public PtNDArray mul(ai.djl.ndarray.NDArray other)
- Specified by:
mulin interfaceai.djl.ndarray.NDArray
-
div
public PtNDArray div(java.lang.Number n)
- Specified by:
divin interfaceai.djl.ndarray.NDArray
-
div
public PtNDArray div(ai.djl.ndarray.NDArray other)
- Specified by:
divin interfaceai.djl.ndarray.NDArray
-
mod
public PtNDArray mod(java.lang.Number n)
- Specified by:
modin interfaceai.djl.ndarray.NDArray
-
mod
public PtNDArray mod(ai.djl.ndarray.NDArray other)
- Specified by:
modin interfaceai.djl.ndarray.NDArray
-
pow
public PtNDArray pow(java.lang.Number n)
- Specified by:
powin interfaceai.djl.ndarray.NDArray
-
pow
public PtNDArray pow(ai.djl.ndarray.NDArray other)
- Specified by:
powin interfaceai.djl.ndarray.NDArray
-
xlogy
public ai.djl.ndarray.NDArray xlogy(ai.djl.ndarray.NDArray other)
- Specified by:
xlogyin interfaceai.djl.ndarray.NDArray
-
addi
public PtNDArray addi(java.lang.Number n)
- Specified by:
addiin interfaceai.djl.ndarray.NDArray
-
addi
public PtNDArray addi(ai.djl.ndarray.NDArray other)
- Specified by:
addiin interfaceai.djl.ndarray.NDArray
-
subi
public PtNDArray subi(java.lang.Number n)
- Specified by:
subiin interfaceai.djl.ndarray.NDArray
-
subi
public PtNDArray subi(ai.djl.ndarray.NDArray other)
- Specified by:
subiin interfaceai.djl.ndarray.NDArray
-
muli
public PtNDArray muli(java.lang.Number n)
- Specified by:
muliin interfaceai.djl.ndarray.NDArray
-
muli
public PtNDArray muli(ai.djl.ndarray.NDArray other)
- Specified by:
muliin interfaceai.djl.ndarray.NDArray
-
divi
public PtNDArray divi(java.lang.Number n)
- Specified by:
diviin interfaceai.djl.ndarray.NDArray
-
divi
public PtNDArray divi(ai.djl.ndarray.NDArray other)
- Specified by:
diviin interfaceai.djl.ndarray.NDArray
-
modi
public PtNDArray modi(java.lang.Number n)
- Specified by:
modiin interfaceai.djl.ndarray.NDArray
-
modi
public PtNDArray modi(ai.djl.ndarray.NDArray other)
- Specified by:
modiin interfaceai.djl.ndarray.NDArray
-
powi
public PtNDArray powi(java.lang.Number n)
- Specified by:
powiin interfaceai.djl.ndarray.NDArray
-
powi
public PtNDArray powi(ai.djl.ndarray.NDArray other)
- Specified by:
powiin interfaceai.djl.ndarray.NDArray
-
sign
public PtNDArray sign()
- Specified by:
signin interfaceai.djl.ndarray.NDArray
-
signi
public PtNDArray signi()
- Specified by:
signiin interfaceai.djl.ndarray.NDArray
-
maximum
public PtNDArray maximum(java.lang.Number n)
- Specified by:
maximumin interfaceai.djl.ndarray.NDArray
-
maximum
public PtNDArray maximum(ai.djl.ndarray.NDArray other)
- Specified by:
maximumin interfaceai.djl.ndarray.NDArray
-
minimum
public PtNDArray minimum(java.lang.Number n)
- Specified by:
minimumin interfaceai.djl.ndarray.NDArray
-
minimum
public PtNDArray minimum(ai.djl.ndarray.NDArray other)
- Specified by:
minimumin interfaceai.djl.ndarray.NDArray
-
all
public PtNDArray all()
- Specified by:
allin interfaceai.djl.ndarray.NDArray
-
any
public PtNDArray any()
- Specified by:
anyin interfaceai.djl.ndarray.NDArray
-
none
public PtNDArray none()
- Specified by:
nonein interfaceai.djl.ndarray.NDArray
-
neg
public PtNDArray neg()
- Specified by:
negin interfaceai.djl.ndarray.NDArray
-
negi
public PtNDArray negi()
- Specified by:
negiin interfaceai.djl.ndarray.NDArray
-
abs
public PtNDArray abs()
- Specified by:
absin interfaceai.djl.ndarray.NDArray
-
square
public PtNDArray square()
- Specified by:
squarein interfaceai.djl.ndarray.NDArray
-
sqrt
public ai.djl.ndarray.NDArray sqrt()
- Specified by:
sqrtin interfaceai.djl.ndarray.NDArray
-
cbrt
public PtNDArray cbrt()
- Specified by:
cbrtin interfaceai.djl.ndarray.NDArray
-
floor
public PtNDArray floor()
- Specified by:
floorin interfaceai.djl.ndarray.NDArray
-
ceil
public PtNDArray ceil()
- Specified by:
ceilin interfaceai.djl.ndarray.NDArray
-
round
public PtNDArray round()
- Specified by:
roundin interfaceai.djl.ndarray.NDArray
-
trunc
public PtNDArray trunc()
- Specified by:
truncin interfaceai.djl.ndarray.NDArray
-
exp
public PtNDArray exp()
- Specified by:
expin interfaceai.djl.ndarray.NDArray
-
gammaln
public ai.djl.ndarray.NDArray gammaln()
- Specified by:
gammalnin interfaceai.djl.ndarray.NDArray
-
log
public PtNDArray log()
- Specified by:
login interfaceai.djl.ndarray.NDArray
-
log10
public PtNDArray log10()
- Specified by:
log10in interfaceai.djl.ndarray.NDArray
-
log2
public PtNDArray log2()
- Specified by:
log2in interfaceai.djl.ndarray.NDArray
-
sin
public PtNDArray sin()
- Specified by:
sinin interfaceai.djl.ndarray.NDArray
-
cos
public PtNDArray cos()
- Specified by:
cosin interfaceai.djl.ndarray.NDArray
-
tan
public PtNDArray tan()
- Specified by:
tanin interfaceai.djl.ndarray.NDArray
-
asin
public PtNDArray asin()
- Specified by:
asinin interfaceai.djl.ndarray.NDArray
-
acos
public PtNDArray acos()
- Specified by:
acosin interfaceai.djl.ndarray.NDArray
-
atan
public PtNDArray atan()
- Specified by:
atanin interfaceai.djl.ndarray.NDArray
-
sinh
public PtNDArray sinh()
- Specified by:
sinhin interfaceai.djl.ndarray.NDArray
-
cosh
public PtNDArray cosh()
- Specified by:
coshin interfaceai.djl.ndarray.NDArray
-
tanh
public PtNDArray tanh()
- Specified by:
tanhin interfaceai.djl.ndarray.NDArray
-
asinh
public PtNDArray asinh()
- Specified by:
asinhin interfaceai.djl.ndarray.NDArray
-
acosh
public PtNDArray acosh()
- Specified by:
acoshin interfaceai.djl.ndarray.NDArray
-
atanh
public PtNDArray atanh()
- Specified by:
atanhin interfaceai.djl.ndarray.NDArray
-
toDegrees
public PtNDArray toDegrees()
- Specified by:
toDegreesin interfaceai.djl.ndarray.NDArray
-
toRadians
public PtNDArray toRadians()
- Specified by:
toRadiansin interfaceai.djl.ndarray.NDArray
-
max
public PtNDArray max()
- Specified by:
maxin interfaceai.djl.ndarray.NDArray
-
max
public PtNDArray max(int[] axes, boolean keepDims)
- Specified by:
maxin interfaceai.djl.ndarray.NDArray
-
min
public PtNDArray min()
- Specified by:
minin interfaceai.djl.ndarray.NDArray
-
min
public PtNDArray min(int[] axes, boolean keepDims)
- Specified by:
minin interfaceai.djl.ndarray.NDArray
-
sum
public PtNDArray sum()
- Specified by:
sumin interfaceai.djl.ndarray.NDArray
-
sum
public PtNDArray sum(int[] axes, boolean keepDims)
- Specified by:
sumin interfaceai.djl.ndarray.NDArray
-
cumProd
public ai.djl.ndarray.NDArray cumProd(int axis)
- Specified by:
cumProdin interfaceai.djl.ndarray.NDArray
-
cumProd
public ai.djl.ndarray.NDArray cumProd(int axis, ai.djl.ndarray.types.DataType dataType)- Specified by:
cumProdin interfaceai.djl.ndarray.NDArray
-
prod
public PtNDArray prod()
- Specified by:
prodin interfaceai.djl.ndarray.NDArray
-
prod
public PtNDArray prod(int[] axes, boolean keepDims)
- Specified by:
prodin interfaceai.djl.ndarray.NDArray
-
mean
public PtNDArray mean()
- Specified by:
meanin interfaceai.djl.ndarray.NDArray
-
mean
public PtNDArray mean(int[] axes, boolean keepDims)
- Specified by:
meanin interfaceai.djl.ndarray.NDArray
-
normalize
public PtNDArray normalize(double p, long dim, double eps)
- Specified by:
normalizein interfaceai.djl.ndarray.NDArray
-
rotate90
public PtNDArray rotate90(int times, int[] axes)
- Specified by:
rotate90in interfaceai.djl.ndarray.NDArray
-
trace
public PtNDArray trace(int offset, int axis1, int axis2)
- Specified by:
tracein interfaceai.djl.ndarray.NDArray
-
split
public ai.djl.ndarray.NDList split(long sections, int axis)- Specified by:
splitin interfaceai.djl.ndarray.NDArray
-
split
public ai.djl.ndarray.NDList split(long[] indices, int axis)- Specified by:
splitin interfaceai.djl.ndarray.NDArray
-
flatten
public PtNDArray flatten()
- Specified by:
flattenin interfaceai.djl.ndarray.NDArray
-
flatten
public ai.djl.ndarray.NDArray flatten(int startDim, int endDim)- Specified by:
flattenin interfaceai.djl.ndarray.NDArray
-
fft
public ai.djl.ndarray.NDArray fft(long length, long axis)- Specified by:
fftin interfaceai.djl.ndarray.NDArray
-
stft
public ai.djl.ndarray.NDArray stft(long nFft, long hopLength, boolean center, ai.djl.ndarray.NDArray window, boolean normalize, boolean returnComplex)- Specified by:
stftin interfaceai.djl.ndarray.NDArray
-
reshape
public PtNDArray reshape(ai.djl.ndarray.types.Shape shape)
- Specified by:
reshapein interfaceai.djl.ndarray.NDArray
-
expandDims
public PtNDArray expandDims(int axis)
- Specified by:
expandDimsin interfaceai.djl.ndarray.NDArray
-
squeeze
public PtNDArray squeeze()
- Specified by:
squeezein interfaceai.djl.ndarray.NDArray
-
squeeze
public PtNDArray squeeze(int axis)
- Specified by:
squeezein interfaceai.djl.ndarray.NDArray
-
squeeze
public PtNDArray squeeze(int[] axes)
- Specified by:
squeezein interfaceai.djl.ndarray.NDArray
-
unique
public ai.djl.ndarray.NDList unique(java.lang.Integer dim, boolean sorted, boolean returnInverse, boolean returnCounts)- Specified by:
uniquein interfaceai.djl.ndarray.NDArray
-
logicalAnd
public PtNDArray logicalAnd(ai.djl.ndarray.NDArray other)
- Specified by:
logicalAndin interfaceai.djl.ndarray.NDArray
-
logicalOr
public PtNDArray logicalOr(ai.djl.ndarray.NDArray other)
- Specified by:
logicalOrin interfaceai.djl.ndarray.NDArray
-
logicalXor
public PtNDArray logicalXor(ai.djl.ndarray.NDArray other)
- Specified by:
logicalXorin interfaceai.djl.ndarray.NDArray
-
logicalNot
public PtNDArray logicalNot()
- Specified by:
logicalNotin interfaceai.djl.ndarray.NDArray
-
argSort
public PtNDArray argSort(int axis, boolean ascending)
- Specified by:
argSortin interfaceai.djl.ndarray.NDArray
-
sort
public PtNDArray sort()
- Specified by:
sortin interfaceai.djl.ndarray.NDArray
-
sort
public PtNDArray sort(int axis)
- Specified by:
sortin interfaceai.djl.ndarray.NDArray
-
softmax
public PtNDArray softmax(int axis)
- Specified by:
softmaxin interfaceai.djl.ndarray.NDArray
-
logSoftmax
public PtNDArray logSoftmax(int axis)
- Specified by:
logSoftmaxin interfaceai.djl.ndarray.NDArray
-
cumSum
public PtNDArray cumSum()
- Specified by:
cumSumin interfaceai.djl.ndarray.NDArray
-
cumSum
public PtNDArray cumSum(int axis)
- Specified by:
cumSumin interfaceai.djl.ndarray.NDArray
-
intern
public void intern(ai.djl.ndarray.NDArray replaced)
- Specified by:
internin interfaceai.djl.ndarray.NDArray
-
isInfinite
public PtNDArray isInfinite()
- Specified by:
isInfinitein interfaceai.djl.ndarray.NDArray
-
isNaN
public PtNDArray isNaN()
- Specified by:
isNaNin interfaceai.djl.ndarray.NDArray
-
tile
public PtNDArray tile(long repeats)
- Specified by:
tilein interfaceai.djl.ndarray.NDArray
-
tile
public PtNDArray tile(int axis, long repeats)
- Specified by:
tilein interfaceai.djl.ndarray.NDArray
-
tile
public PtNDArray tile(long[] repeats)
- Specified by:
tilein interfaceai.djl.ndarray.NDArray
-
tile
public PtNDArray tile(ai.djl.ndarray.types.Shape desiredShape)
- Specified by:
tilein interfaceai.djl.ndarray.NDArray
-
repeat
public PtNDArray repeat(long repeats)
- Specified by:
repeatin interfaceai.djl.ndarray.NDArray
-
repeat
public PtNDArray repeat(int axis, long repeats)
- Specified by:
repeatin interfaceai.djl.ndarray.NDArray
-
repeat
public PtNDArray repeat(long[] repeats)
- Specified by:
repeatin interfaceai.djl.ndarray.NDArray
-
repeat
public PtNDArray repeat(ai.djl.ndarray.types.Shape desiredShape)
- Specified by:
repeatin interfaceai.djl.ndarray.NDArray
-
dot
public PtNDArray dot(ai.djl.ndarray.NDArray other)
- Specified by:
dotin interfaceai.djl.ndarray.NDArray
-
matMul
public ai.djl.ndarray.NDArray matMul(ai.djl.ndarray.NDArray other)
- Specified by:
matMulin interfaceai.djl.ndarray.NDArray
-
batchMatMul
public ai.djl.ndarray.NDArray batchMatMul(ai.djl.ndarray.NDArray other)
- Specified by:
batchMatMulin interfaceai.djl.ndarray.NDArray
-
clip
public PtNDArray clip(java.lang.Number min, java.lang.Number max)
- Specified by:
clipin interfaceai.djl.ndarray.NDArray
-
swapAxes
public PtNDArray swapAxes(int axis1, int axis2)
- Specified by:
swapAxesin interfaceai.djl.ndarray.NDArray
-
flip
public ai.djl.ndarray.NDArray flip(int... axes)
- Specified by:
flipin interfaceai.djl.ndarray.NDArray
-
transpose
public PtNDArray transpose()
- Specified by:
transposein interfaceai.djl.ndarray.NDArray
-
transpose
public PtNDArray transpose(int... axes)
- Specified by:
transposein interfaceai.djl.ndarray.NDArray
-
broadcast
public PtNDArray broadcast(ai.djl.ndarray.types.Shape shape)
- Specified by:
broadcastin interfaceai.djl.ndarray.NDArray
-
argMax
public PtNDArray argMax()
- Specified by:
argMaxin interfaceai.djl.ndarray.NDArray
-
argMax
public PtNDArray argMax(int axis)
- Specified by:
argMaxin interfaceai.djl.ndarray.NDArray
-
topK
public ai.djl.ndarray.NDList topK(int k, int axis, boolean largest, boolean sorted)- Specified by:
topKin interfaceai.djl.ndarray.NDArray
-
argMin
public PtNDArray argMin()
- Specified by:
argMinin interfaceai.djl.ndarray.NDArray
-
argMin
public PtNDArray argMin(int axis)
- Specified by:
argMinin interfaceai.djl.ndarray.NDArray
-
percentile
public PtNDArray percentile(java.lang.Number percentile)
- Specified by:
percentilein interfaceai.djl.ndarray.NDArray
-
percentile
public PtNDArray percentile(java.lang.Number percentile, int[] axes)
- Specified by:
percentilein interfaceai.djl.ndarray.NDArray
-
median
public PtNDArray median()
- Specified by:
medianin interfaceai.djl.ndarray.NDArray
-
median
public PtNDArray median(int[] axes)
- Specified by:
medianin interfaceai.djl.ndarray.NDArray
-
toDense
public PtNDArray toDense()
- Specified by:
toDensein interfaceai.djl.ndarray.NDArray
-
toSparse
public PtNDArray toSparse(ai.djl.ndarray.types.SparseFormat fmt)
- Specified by:
toSparsein interfaceai.djl.ndarray.NDArray
-
nonzero
public PtNDArray nonzero()
- Specified by:
nonzeroin interfaceai.djl.ndarray.NDArray
-
erfinv
public PtNDArray erfinv()
- Specified by:
erfinvin interfaceai.djl.ndarray.NDArray
-
inverse
public PtNDArray inverse()
- Specified by:
inversein interfaceai.djl.ndarray.NDArray
-
norm
public ai.djl.ndarray.NDArray norm(boolean keepDims)
- Specified by:
normin interfaceai.djl.ndarray.NDArray
-
norm
public ai.djl.ndarray.NDArray norm(int order, int[] axes, boolean keepDims)- Specified by:
normin interfaceai.djl.ndarray.NDArray
-
oneHot
public ai.djl.ndarray.NDArray oneHot(int depth)
- Specified by:
oneHotin interfaceai.djl.ndarray.NDArray
-
oneHot
public ai.djl.ndarray.NDArray oneHot(int depth, ai.djl.ndarray.types.DataType dataType)- Specified by:
oneHotin interfaceai.djl.ndarray.NDArray
-
oneHot
public ai.djl.ndarray.NDArray oneHot(int depth, float onValue, float offValue, ai.djl.ndarray.types.DataType dataType)- Specified by:
oneHotin interfaceai.djl.ndarray.NDArray
-
batchDot
public ai.djl.ndarray.NDArray batchDot(ai.djl.ndarray.NDArray other)
- Specified by:
batchDotin interfaceai.djl.ndarray.NDArray
-
complex
public ai.djl.ndarray.NDArray complex()
- Specified by:
complexin interfaceai.djl.ndarray.NDArray
-
real
public ai.djl.ndarray.NDArray real()
- Specified by:
realin interfaceai.djl.ndarray.NDArray
-
getNDArrayInternal
public PtNDArrayEx getNDArrayInternal()
- Specified by:
getNDArrayInternalin interfaceai.djl.ndarray.NDArray
-
toString
public java.lang.String toString()
- Overrides:
toStringin classjava.lang.Object
-
equals
public boolean equals(java.lang.Object obj)
- Overrides:
equalsin classjava.lang.Object
-
hashCode
public int hashCode()
- Overrides:
hashCodein classjava.lang.Object
-
close
public void close()
- Specified by:
closein interfacejava.lang.AutoCloseable- Specified by:
closein interfaceai.djl.ndarray.NDArray- Specified by:
closein interfaceai.djl.ndarray.NDResource- Overrides:
closein classai.djl.util.NativeResource<java.lang.Long>
-
-