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.
-
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)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()booleancontentEquals(ai.djl.ndarray.NDArray other)booleancontentEquals(java.lang.Number number)voidcopyTo(ai.djl.ndarray.NDArray array)PtNDArraycos()PtNDArraycosh()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)PtNDArrayflatten()ai.djl.ndarray.NDArrayflip(int... axes)PtNDArrayfloor()ai.djl.ndarray.NDArraygather(ai.djl.ndarray.NDArray index, int axis)ai.djl.ndarray.NDArrayget(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)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)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)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)PtNDArrayrotate90(int times, int[] axes)PtNDArrayround()ai.djl.ndarray.NDArraysequenceMask(ai.djl.ndarray.NDArray sequenceLength)ai.djl.ndarray.NDArraysequenceMask(ai.djl.ndarray.NDArray sequenceLength, float value)voidset(java.nio.Buffer data)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.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)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)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()-
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, countNonzero, countNonzero, encode, get, get, get, getBoolean, getByte, getDouble, getFloat, getInt, getLong, getScalar, getUid, getUint8, isEmpty, isScalar, isSparse, like, max, mean, min, norm, norm, norm, onesLike, prod, reshape, scaleGradient, set, set, set, set, set, set, set, set, set, setScalar, shapeEquals, size, size, split, split, stack, stack, sum, toArray, toBooleanArray, toByteArray, toDebugString, toDebugString, toDoubleArray, toFloatArray, toIntArray, toLongArray, toStringArray, toUint8Array, trace, trace, 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
-
-
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 data)
- Specified by:
setin interfaceai.djl.ndarray.NDArray
-
get
public ai.djl.ndarray.NDArray get(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
-
copyTo
public void copyTo(ai.djl.ndarray.NDArray array)
- Specified by:
copyToin interfaceai.djl.ndarray.NDArray
-
attach
public void attach(ai.djl.ndarray.NDManager manager)
- Specified by:
attachin 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
-
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
-
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
-
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
-
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
-
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
-
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
-
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
-
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
-
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
-
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>
-
-