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.NDArray
PtNDArray
is the PyTorch implementation ofNDArray
.
-
-
Constructor Summary
Constructors Constructor Description PtNDArray(PtNDManager manager, long handle)
Constructs a PyTorchNDArray
from a native handle (internal.PtNDArray(PtNDManager manager, long handle, java.nio.ByteBuffer data)
Constructs a PyTorchNDArray
from a native handle (internal.PtNDArray(PtNDManager manager, java.lang.String[] strs, ai.djl.ndarray.types.Shape shape)
Constructs a PyTorchNDArray
to hold string array with a dummy native handle (internal.
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description PtNDArray
abs()
PtNDArray
acos()
PtNDArray
acosh()
PtNDArray
add(ai.djl.ndarray.NDArray other)
PtNDArray
add(java.lang.Number n)
PtNDArray
addi(ai.djl.ndarray.NDArray other)
PtNDArray
addi(java.lang.Number n)
PtNDArray
all()
PtNDArray
any()
PtNDArray
argMax()
PtNDArray
argMax(int axis)
PtNDArray
argMin()
PtNDArray
argMin(int axis)
PtNDArray
argSort(int axis, boolean ascending)
PtNDArray
asin()
PtNDArray
asinh()
PtNDArray
atan()
PtNDArray
atanh()
void
attach(ai.djl.ndarray.NDManager manager)
ai.djl.ndarray.NDArray
batchDot(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
batchMatMul(ai.djl.ndarray.NDArray other)
PtNDArray
booleanMask(ai.djl.ndarray.NDArray index, int axis)
PtNDArray
broadcast(ai.djl.ndarray.types.Shape shape)
PtNDArray
cbrt()
PtNDArray
ceil()
PtNDArray
clip(java.lang.Number min, java.lang.Number max)
void
close()
ai.djl.ndarray.NDArray
complex()
boolean
contentEquals(ai.djl.ndarray.NDArray other)
boolean
contentEquals(java.lang.Number number)
PtNDArray
cos()
PtNDArray
cosh()
ai.djl.ndarray.NDArray
cumProd(int axis)
ai.djl.ndarray.NDArray
cumProd(int axis, ai.djl.ndarray.types.DataType dataType)
PtNDArray
cumSum()
PtNDArray
cumSum(int axis)
void
detach()
PtNDArray
div(ai.djl.ndarray.NDArray other)
PtNDArray
div(java.lang.Number n)
PtNDArray
divi(ai.djl.ndarray.NDArray other)
PtNDArray
divi(java.lang.Number n)
PtNDArray
dot(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
duplicate()
PtNDArray
eq(ai.djl.ndarray.NDArray other)
PtNDArray
eq(java.lang.Number n)
boolean
equals(java.lang.Object obj)
PtNDArray
erfinv()
PtNDArray
exp()
PtNDArray
expandDims(int axis)
ai.djl.ndarray.NDArray
fft(long length, long axis)
PtNDArray
flatten()
ai.djl.ndarray.NDArray
flatten(int startDim, int endDim)
ai.djl.ndarray.NDArray
flip(int... axes)
PtNDArray
floor()
ai.djl.ndarray.NDArray
gammaln()
ai.djl.ndarray.NDArray
gather(ai.djl.ndarray.NDArray index, int axis)
ai.djl.ndarray.NDArray
gatherNd(ai.djl.ndarray.NDArray index)
ai.djl.ndarray.NDArray
get(ai.djl.ndarray.NDManager manager, long... indices)
ai.djl.ndarray.types.DataType
getDataType()
ai.djl.Device
getDevice()
PtNDArray
getGradient()
PtNDManager
getManager()
java.lang.String
getName()
PtNDArrayEx
getNDArrayInternal()
ai.djl.ndarray.types.Shape
getShape()
ai.djl.ndarray.types.SparseFormat
getSparseFormat()
PtNDArray
gt(ai.djl.ndarray.NDArray other)
PtNDArray
gt(java.lang.Number n)
PtNDArray
gte(ai.djl.ndarray.NDArray other)
PtNDArray
gte(java.lang.Number n)
boolean
hasGradient()
int
hashCode()
void
intern(ai.djl.ndarray.NDArray replaced)
PtNDArray
inverse()
PtNDArray
isInfinite()
PtNDArray
isNaN()
PtNDArray
log()
PtNDArray
log10()
PtNDArray
log2()
PtNDArray
logicalAnd(ai.djl.ndarray.NDArray other)
PtNDArray
logicalNot()
PtNDArray
logicalOr(ai.djl.ndarray.NDArray other)
PtNDArray
logicalXor(ai.djl.ndarray.NDArray other)
PtNDArray
logSoftmax(int axis)
PtNDArray
lt(ai.djl.ndarray.NDArray other)
PtNDArray
lt(java.lang.Number n)
PtNDArray
lte(ai.djl.ndarray.NDArray other)
PtNDArray
lte(java.lang.Number n)
ai.djl.ndarray.NDArray
matMul(ai.djl.ndarray.NDArray other)
PtNDArray
max()
PtNDArray
max(int[] axes, boolean keepDims)
PtNDArray
maximum(ai.djl.ndarray.NDArray other)
PtNDArray
maximum(java.lang.Number n)
PtNDArray
mean()
PtNDArray
mean(int[] axes, boolean keepDims)
PtNDArray
median()
PtNDArray
median(int[] axes)
PtNDArray
min()
PtNDArray
min(int[] axes, boolean keepDims)
PtNDArray
minimum(ai.djl.ndarray.NDArray other)
PtNDArray
minimum(java.lang.Number n)
PtNDArray
mod(ai.djl.ndarray.NDArray other)
PtNDArray
mod(java.lang.Number n)
PtNDArray
modi(ai.djl.ndarray.NDArray other)
PtNDArray
modi(java.lang.Number n)
PtNDArray
mul(ai.djl.ndarray.NDArray other)
PtNDArray
mul(java.lang.Number n)
PtNDArray
muli(ai.djl.ndarray.NDArray other)
PtNDArray
muli(java.lang.Number n)
PtNDArray
neg()
PtNDArray
negi()
PtNDArray
neq(ai.djl.ndarray.NDArray other)
PtNDArray
neq(java.lang.Number n)
PtNDArray
none()
PtNDArray
nonzero()
ai.djl.ndarray.NDArray
norm(boolean keepDims)
ai.djl.ndarray.NDArray
norm(int order, int[] axes, boolean keepDims)
PtNDArray
normalize(double p, long dim, double eps)
ai.djl.ndarray.NDArray
oneHot(int depth)
ai.djl.ndarray.NDArray
oneHot(int depth, float onValue, float offValue, ai.djl.ndarray.types.DataType dataType)
ai.djl.ndarray.NDArray
oneHot(int depth, ai.djl.ndarray.types.DataType dataType)
PtNDArray
percentile(java.lang.Number percentile)
PtNDArray
percentile(java.lang.Number percentile, int[] axes)
PtNDArray
pow(ai.djl.ndarray.NDArray other)
PtNDArray
pow(java.lang.Number n)
PtNDArray
powi(ai.djl.ndarray.NDArray other)
PtNDArray
powi(java.lang.Number n)
PtNDArray
prod()
PtNDArray
prod(int[] axes, boolean keepDims)
ai.djl.ndarray.NDArray
put(ai.djl.ndarray.NDArray index, ai.djl.ndarray.NDArray value)
ai.djl.ndarray.NDArray
real()
PtNDArray
repeat(int axis, long repeats)
PtNDArray
repeat(long repeats)
PtNDArray
repeat(long[] repeats)
PtNDArray
repeat(ai.djl.ndarray.types.Shape desiredShape)
PtNDArray
reshape(ai.djl.ndarray.types.Shape shape)
void
returnResource(ai.djl.ndarray.NDManager manager)
PtNDArray
rotate90(int times, int[] axes)
PtNDArray
round()
ai.djl.ndarray.NDArray
scatter(ai.djl.ndarray.NDArray index, ai.djl.ndarray.NDArray value, int axis)
ai.djl.ndarray.NDArray
sequenceMask(ai.djl.ndarray.NDArray sequenceLength)
ai.djl.ndarray.NDArray
sequenceMask(ai.djl.ndarray.NDArray sequenceLength, float value)
void
set(java.nio.Buffer buffer)
void
setName(java.lang.String name)
void
setRequiresGradient(boolean requiresGrad)
PtNDArray
sign()
PtNDArray
signi()
PtNDArray
sin()
PtNDArray
sinh()
PtNDArray
softmax(int axis)
PtNDArray
sort()
PtNDArray
sort(int axis)
ai.djl.ndarray.NDList
split(long[] indices, int axis)
ai.djl.ndarray.NDList
split(long sections, int axis)
ai.djl.ndarray.NDArray
sqrt()
PtNDArray
square()
PtNDArray
squeeze()
PtNDArray
squeeze(int axis)
PtNDArray
squeeze(int[] axes)
ai.djl.ndarray.NDArray
stft(long nFft, long hopLength, boolean center, ai.djl.ndarray.NDArray window, boolean normalize, boolean returnComplex)
ai.djl.ndarray.NDArray
stopGradient()
PtNDArray
sub(ai.djl.ndarray.NDArray other)
PtNDArray
sub(java.lang.Number n)
PtNDArray
subi(ai.djl.ndarray.NDArray other)
PtNDArray
subi(java.lang.Number n)
PtNDArray
sum()
PtNDArray
sum(int[] axes, boolean keepDims)
PtNDArray
swapAxes(int axis1, int axis2)
ai.djl.ndarray.NDArray
take(ai.djl.ndarray.NDManager manager, ai.djl.ndarray.NDArray index)
PtNDArray
tan()
PtNDArray
tanh()
void
tempAttach(ai.djl.ndarray.NDManager manager)
PtNDArray
tile(int axis, long repeats)
PtNDArray
tile(long repeats)
PtNDArray
tile(long[] repeats)
PtNDArray
tile(ai.djl.ndarray.types.Shape desiredShape)
java.nio.ByteBuffer
toByteBuffer()
PtNDArray
toDegrees()
PtNDArray
toDense()
PtNDArray
toDevice(ai.djl.Device device, boolean copy)
ai.djl.ndarray.NDList
topK(int k, int axis, boolean largest, boolean sorted)
PtNDArray
toRadians()
PtNDArray
toSparse(ai.djl.ndarray.types.SparseFormat fmt)
java.lang.String
toString()
java.lang.String[]
toStringArray(java.nio.charset.Charset charset)
PtNDArray
toType(ai.djl.ndarray.types.DataType dataType, boolean copy)
PtNDArray
trace(int offset, int axis1, int axis2)
PtNDArray
transpose()
PtNDArray
transpose(int... axes)
PtNDArray
trunc()
ai.djl.ndarray.NDList
unique(java.lang.Integer dim, boolean sorted, boolean returnInverse, boolean returnCounts)
ai.djl.ndarray.NDArray
xlogy(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 PyTorchNDArray
from a native handle (internal. UseNDManager
instead).- 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 PyTorchNDArray
from a native handle (internal. UseNDManager
instead) 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 PyTorchNDArray
to hold string array with a dummy native handle (internal. UseNDManager
instead) with the data that is hold on Java side.- Parameters:
manager
- the manager to attach the new array tostrs
- the string arrayshape
- theShape
of theNDArray
-
-
Method Detail
-
getManager
public PtNDManager getManager()
- Specified by:
getManager
in interfaceai.djl.ndarray.NDResource
-
getName
public java.lang.String getName()
- Specified by:
getName
in interfaceai.djl.ndarray.NDArray
-
setName
public void setName(java.lang.String name)
- Specified by:
setName
in interfaceai.djl.ndarray.NDArray
-
getDataType
public ai.djl.ndarray.types.DataType getDataType()
- Specified by:
getDataType
in interfaceai.djl.ndarray.NDArray
-
getDevice
public ai.djl.Device getDevice()
- Specified by:
getDevice
in interfaceai.djl.ndarray.NDArray
-
getShape
public ai.djl.ndarray.types.Shape getShape()
- Specified by:
getShape
in interfaceai.djl.ndarray.NDArray
-
getSparseFormat
public ai.djl.ndarray.types.SparseFormat getSparseFormat()
- Specified by:
getSparseFormat
in interfaceai.djl.ndarray.NDArray
-
toDevice
public PtNDArray toDevice(ai.djl.Device device, boolean copy)
- Specified by:
toDevice
in interfaceai.djl.ndarray.NDArray
-
toType
public PtNDArray toType(ai.djl.ndarray.types.DataType dataType, boolean copy)
- Specified by:
toType
in interfaceai.djl.ndarray.NDArray
-
setRequiresGradient
public void setRequiresGradient(boolean requiresGrad)
- Specified by:
setRequiresGradient
in interfaceai.djl.ndarray.NDArray
-
getGradient
public PtNDArray getGradient()
- Specified by:
getGradient
in interfaceai.djl.ndarray.NDArray
-
hasGradient
public boolean hasGradient()
- Specified by:
hasGradient
in interfaceai.djl.ndarray.NDArray
-
stopGradient
public ai.djl.ndarray.NDArray stopGradient()
- Specified by:
stopGradient
in interfaceai.djl.ndarray.NDArray
-
toByteBuffer
public java.nio.ByteBuffer toByteBuffer()
- Specified by:
toByteBuffer
in interfaceai.djl.ndarray.BytesSupplier
-
toStringArray
public java.lang.String[] toStringArray(java.nio.charset.Charset charset)
- Specified by:
toStringArray
in interfaceai.djl.ndarray.NDArray
-
set
public void set(java.nio.Buffer buffer)
- Specified by:
set
in interfaceai.djl.ndarray.NDArray
-
get
public ai.djl.ndarray.NDArray get(ai.djl.ndarray.NDManager manager, long... indices)
- Specified by:
get
in interfaceai.djl.ndarray.NDArray
-
gather
public ai.djl.ndarray.NDArray gather(ai.djl.ndarray.NDArray index, int axis)
- Specified by:
gather
in interfaceai.djl.ndarray.NDArray
-
gatherNd
public ai.djl.ndarray.NDArray gatherNd(ai.djl.ndarray.NDArray index)
- Specified by:
gatherNd
in interfaceai.djl.ndarray.NDArray
-
take
public ai.djl.ndarray.NDArray take(ai.djl.ndarray.NDManager manager, ai.djl.ndarray.NDArray index)
- Specified by:
take
in interfaceai.djl.ndarray.NDArray
-
put
public ai.djl.ndarray.NDArray put(ai.djl.ndarray.NDArray index, ai.djl.ndarray.NDArray value)
- Specified by:
put
in 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:
scatter
in interfaceai.djl.ndarray.NDArray
-
attach
public void attach(ai.djl.ndarray.NDManager manager)
- Specified by:
attach
in interfaceai.djl.ndarray.NDResource
-
returnResource
public void returnResource(ai.djl.ndarray.NDManager manager)
- Specified by:
returnResource
in interfaceai.djl.ndarray.NDResource
-
tempAttach
public void tempAttach(ai.djl.ndarray.NDManager manager)
- Specified by:
tempAttach
in interfaceai.djl.ndarray.NDResource
-
detach
public void detach()
- Specified by:
detach
in interfaceai.djl.ndarray.NDResource
-
duplicate
public ai.djl.ndarray.NDArray duplicate()
- Specified by:
duplicate
in interfaceai.djl.ndarray.NDArray
-
booleanMask
public PtNDArray booleanMask(ai.djl.ndarray.NDArray index, int axis)
- Specified by:
booleanMask
in interfaceai.djl.ndarray.NDArray
-
sequenceMask
public ai.djl.ndarray.NDArray sequenceMask(ai.djl.ndarray.NDArray sequenceLength, float value)
- Specified by:
sequenceMask
in interfaceai.djl.ndarray.NDArray
-
sequenceMask
public ai.djl.ndarray.NDArray sequenceMask(ai.djl.ndarray.NDArray sequenceLength)
- Specified by:
sequenceMask
in interfaceai.djl.ndarray.NDArray
-
contentEquals
public boolean contentEquals(java.lang.Number number)
- Specified by:
contentEquals
in interfaceai.djl.ndarray.NDArray
-
contentEquals
public boolean contentEquals(ai.djl.ndarray.NDArray other)
- Specified by:
contentEquals
in interfaceai.djl.ndarray.NDArray
-
eq
public PtNDArray eq(java.lang.Number n)
- Specified by:
eq
in interfaceai.djl.ndarray.NDArray
-
eq
public PtNDArray eq(ai.djl.ndarray.NDArray other)
- Specified by:
eq
in interfaceai.djl.ndarray.NDArray
-
neq
public PtNDArray neq(java.lang.Number n)
- Specified by:
neq
in interfaceai.djl.ndarray.NDArray
-
neq
public PtNDArray neq(ai.djl.ndarray.NDArray other)
- Specified by:
neq
in interfaceai.djl.ndarray.NDArray
-
gt
public PtNDArray gt(java.lang.Number n)
- Specified by:
gt
in interfaceai.djl.ndarray.NDArray
-
gt
public PtNDArray gt(ai.djl.ndarray.NDArray other)
- Specified by:
gt
in interfaceai.djl.ndarray.NDArray
-
gte
public PtNDArray gte(java.lang.Number n)
- Specified by:
gte
in interfaceai.djl.ndarray.NDArray
-
gte
public PtNDArray gte(ai.djl.ndarray.NDArray other)
- Specified by:
gte
in interfaceai.djl.ndarray.NDArray
-
lt
public PtNDArray lt(java.lang.Number n)
- Specified by:
lt
in interfaceai.djl.ndarray.NDArray
-
lt
public PtNDArray lt(ai.djl.ndarray.NDArray other)
- Specified by:
lt
in interfaceai.djl.ndarray.NDArray
-
lte
public PtNDArray lte(java.lang.Number n)
- Specified by:
lte
in interfaceai.djl.ndarray.NDArray
-
lte
public PtNDArray lte(ai.djl.ndarray.NDArray other)
- Specified by:
lte
in interfaceai.djl.ndarray.NDArray
-
add
public PtNDArray add(java.lang.Number n)
- Specified by:
add
in interfaceai.djl.ndarray.NDArray
-
add
public PtNDArray add(ai.djl.ndarray.NDArray other)
- Specified by:
add
in interfaceai.djl.ndarray.NDArray
-
sub
public PtNDArray sub(java.lang.Number n)
- Specified by:
sub
in interfaceai.djl.ndarray.NDArray
-
sub
public PtNDArray sub(ai.djl.ndarray.NDArray other)
- Specified by:
sub
in interfaceai.djl.ndarray.NDArray
-
mul
public PtNDArray mul(java.lang.Number n)
- Specified by:
mul
in interfaceai.djl.ndarray.NDArray
-
mul
public PtNDArray mul(ai.djl.ndarray.NDArray other)
- Specified by:
mul
in interfaceai.djl.ndarray.NDArray
-
div
public PtNDArray div(java.lang.Number n)
- Specified by:
div
in interfaceai.djl.ndarray.NDArray
-
div
public PtNDArray div(ai.djl.ndarray.NDArray other)
- Specified by:
div
in interfaceai.djl.ndarray.NDArray
-
mod
public PtNDArray mod(java.lang.Number n)
- Specified by:
mod
in interfaceai.djl.ndarray.NDArray
-
mod
public PtNDArray mod(ai.djl.ndarray.NDArray other)
- Specified by:
mod
in interfaceai.djl.ndarray.NDArray
-
pow
public PtNDArray pow(java.lang.Number n)
- Specified by:
pow
in interfaceai.djl.ndarray.NDArray
-
pow
public PtNDArray pow(ai.djl.ndarray.NDArray other)
- Specified by:
pow
in interfaceai.djl.ndarray.NDArray
-
xlogy
public ai.djl.ndarray.NDArray xlogy(ai.djl.ndarray.NDArray other)
- Specified by:
xlogy
in interfaceai.djl.ndarray.NDArray
-
addi
public PtNDArray addi(java.lang.Number n)
- Specified by:
addi
in interfaceai.djl.ndarray.NDArray
-
addi
public PtNDArray addi(ai.djl.ndarray.NDArray other)
- Specified by:
addi
in interfaceai.djl.ndarray.NDArray
-
subi
public PtNDArray subi(java.lang.Number n)
- Specified by:
subi
in interfaceai.djl.ndarray.NDArray
-
subi
public PtNDArray subi(ai.djl.ndarray.NDArray other)
- Specified by:
subi
in interfaceai.djl.ndarray.NDArray
-
muli
public PtNDArray muli(java.lang.Number n)
- Specified by:
muli
in interfaceai.djl.ndarray.NDArray
-
muli
public PtNDArray muli(ai.djl.ndarray.NDArray other)
- Specified by:
muli
in interfaceai.djl.ndarray.NDArray
-
divi
public PtNDArray divi(java.lang.Number n)
- Specified by:
divi
in interfaceai.djl.ndarray.NDArray
-
divi
public PtNDArray divi(ai.djl.ndarray.NDArray other)
- Specified by:
divi
in interfaceai.djl.ndarray.NDArray
-
modi
public PtNDArray modi(java.lang.Number n)
- Specified by:
modi
in interfaceai.djl.ndarray.NDArray
-
modi
public PtNDArray modi(ai.djl.ndarray.NDArray other)
- Specified by:
modi
in interfaceai.djl.ndarray.NDArray
-
powi
public PtNDArray powi(java.lang.Number n)
- Specified by:
powi
in interfaceai.djl.ndarray.NDArray
-
powi
public PtNDArray powi(ai.djl.ndarray.NDArray other)
- Specified by:
powi
in interfaceai.djl.ndarray.NDArray
-
sign
public PtNDArray sign()
- Specified by:
sign
in interfaceai.djl.ndarray.NDArray
-
signi
public PtNDArray signi()
- Specified by:
signi
in interfaceai.djl.ndarray.NDArray
-
maximum
public PtNDArray maximum(java.lang.Number n)
- Specified by:
maximum
in interfaceai.djl.ndarray.NDArray
-
maximum
public PtNDArray maximum(ai.djl.ndarray.NDArray other)
- Specified by:
maximum
in interfaceai.djl.ndarray.NDArray
-
minimum
public PtNDArray minimum(java.lang.Number n)
- Specified by:
minimum
in interfaceai.djl.ndarray.NDArray
-
minimum
public PtNDArray minimum(ai.djl.ndarray.NDArray other)
- Specified by:
minimum
in interfaceai.djl.ndarray.NDArray
-
all
public PtNDArray all()
- Specified by:
all
in interfaceai.djl.ndarray.NDArray
-
any
public PtNDArray any()
- Specified by:
any
in interfaceai.djl.ndarray.NDArray
-
none
public PtNDArray none()
- Specified by:
none
in interfaceai.djl.ndarray.NDArray
-
neg
public PtNDArray neg()
- Specified by:
neg
in interfaceai.djl.ndarray.NDArray
-
negi
public PtNDArray negi()
- Specified by:
negi
in interfaceai.djl.ndarray.NDArray
-
abs
public PtNDArray abs()
- Specified by:
abs
in interfaceai.djl.ndarray.NDArray
-
square
public PtNDArray square()
- Specified by:
square
in interfaceai.djl.ndarray.NDArray
-
sqrt
public ai.djl.ndarray.NDArray sqrt()
- Specified by:
sqrt
in interfaceai.djl.ndarray.NDArray
-
cbrt
public PtNDArray cbrt()
- Specified by:
cbrt
in interfaceai.djl.ndarray.NDArray
-
floor
public PtNDArray floor()
- Specified by:
floor
in interfaceai.djl.ndarray.NDArray
-
ceil
public PtNDArray ceil()
- Specified by:
ceil
in interfaceai.djl.ndarray.NDArray
-
round
public PtNDArray round()
- Specified by:
round
in interfaceai.djl.ndarray.NDArray
-
trunc
public PtNDArray trunc()
- Specified by:
trunc
in interfaceai.djl.ndarray.NDArray
-
exp
public PtNDArray exp()
- Specified by:
exp
in interfaceai.djl.ndarray.NDArray
-
gammaln
public ai.djl.ndarray.NDArray gammaln()
- Specified by:
gammaln
in interfaceai.djl.ndarray.NDArray
-
log
public PtNDArray log()
- Specified by:
log
in interfaceai.djl.ndarray.NDArray
-
log10
public PtNDArray log10()
- Specified by:
log10
in interfaceai.djl.ndarray.NDArray
-
log2
public PtNDArray log2()
- Specified by:
log2
in interfaceai.djl.ndarray.NDArray
-
sin
public PtNDArray sin()
- Specified by:
sin
in interfaceai.djl.ndarray.NDArray
-
cos
public PtNDArray cos()
- Specified by:
cos
in interfaceai.djl.ndarray.NDArray
-
tan
public PtNDArray tan()
- Specified by:
tan
in interfaceai.djl.ndarray.NDArray
-
asin
public PtNDArray asin()
- Specified by:
asin
in interfaceai.djl.ndarray.NDArray
-
acos
public PtNDArray acos()
- Specified by:
acos
in interfaceai.djl.ndarray.NDArray
-
atan
public PtNDArray atan()
- Specified by:
atan
in interfaceai.djl.ndarray.NDArray
-
sinh
public PtNDArray sinh()
- Specified by:
sinh
in interfaceai.djl.ndarray.NDArray
-
cosh
public PtNDArray cosh()
- Specified by:
cosh
in interfaceai.djl.ndarray.NDArray
-
tanh
public PtNDArray tanh()
- Specified by:
tanh
in interfaceai.djl.ndarray.NDArray
-
asinh
public PtNDArray asinh()
- Specified by:
asinh
in interfaceai.djl.ndarray.NDArray
-
acosh
public PtNDArray acosh()
- Specified by:
acosh
in interfaceai.djl.ndarray.NDArray
-
atanh
public PtNDArray atanh()
- Specified by:
atanh
in interfaceai.djl.ndarray.NDArray
-
toDegrees
public PtNDArray toDegrees()
- Specified by:
toDegrees
in interfaceai.djl.ndarray.NDArray
-
toRadians
public PtNDArray toRadians()
- Specified by:
toRadians
in interfaceai.djl.ndarray.NDArray
-
max
public PtNDArray max()
- Specified by:
max
in interfaceai.djl.ndarray.NDArray
-
max
public PtNDArray max(int[] axes, boolean keepDims)
- Specified by:
max
in interfaceai.djl.ndarray.NDArray
-
min
public PtNDArray min()
- Specified by:
min
in interfaceai.djl.ndarray.NDArray
-
min
public PtNDArray min(int[] axes, boolean keepDims)
- Specified by:
min
in interfaceai.djl.ndarray.NDArray
-
sum
public PtNDArray sum()
- Specified by:
sum
in interfaceai.djl.ndarray.NDArray
-
sum
public PtNDArray sum(int[] axes, boolean keepDims)
- Specified by:
sum
in interfaceai.djl.ndarray.NDArray
-
cumProd
public ai.djl.ndarray.NDArray cumProd(int axis)
- Specified by:
cumProd
in interfaceai.djl.ndarray.NDArray
-
cumProd
public ai.djl.ndarray.NDArray cumProd(int axis, ai.djl.ndarray.types.DataType dataType)
- Specified by:
cumProd
in interfaceai.djl.ndarray.NDArray
-
prod
public PtNDArray prod()
- Specified by:
prod
in interfaceai.djl.ndarray.NDArray
-
prod
public PtNDArray prod(int[] axes, boolean keepDims)
- Specified by:
prod
in interfaceai.djl.ndarray.NDArray
-
mean
public PtNDArray mean()
- Specified by:
mean
in interfaceai.djl.ndarray.NDArray
-
mean
public PtNDArray mean(int[] axes, boolean keepDims)
- Specified by:
mean
in interfaceai.djl.ndarray.NDArray
-
normalize
public PtNDArray normalize(double p, long dim, double eps)
- Specified by:
normalize
in interfaceai.djl.ndarray.NDArray
-
rotate90
public PtNDArray rotate90(int times, int[] axes)
- Specified by:
rotate90
in interfaceai.djl.ndarray.NDArray
-
trace
public PtNDArray trace(int offset, int axis1, int axis2)
- Specified by:
trace
in interfaceai.djl.ndarray.NDArray
-
split
public ai.djl.ndarray.NDList split(long sections, int axis)
- Specified by:
split
in interfaceai.djl.ndarray.NDArray
-
split
public ai.djl.ndarray.NDList split(long[] indices, int axis)
- Specified by:
split
in interfaceai.djl.ndarray.NDArray
-
flatten
public PtNDArray flatten()
- Specified by:
flatten
in interfaceai.djl.ndarray.NDArray
-
flatten
public ai.djl.ndarray.NDArray flatten(int startDim, int endDim)
- Specified by:
flatten
in interfaceai.djl.ndarray.NDArray
-
fft
public ai.djl.ndarray.NDArray fft(long length, long axis)
- Specified by:
fft
in 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:
stft
in interfaceai.djl.ndarray.NDArray
-
reshape
public PtNDArray reshape(ai.djl.ndarray.types.Shape shape)
- Specified by:
reshape
in interfaceai.djl.ndarray.NDArray
-
expandDims
public PtNDArray expandDims(int axis)
- Specified by:
expandDims
in interfaceai.djl.ndarray.NDArray
-
squeeze
public PtNDArray squeeze()
- Specified by:
squeeze
in interfaceai.djl.ndarray.NDArray
-
squeeze
public PtNDArray squeeze(int axis)
- Specified by:
squeeze
in interfaceai.djl.ndarray.NDArray
-
squeeze
public PtNDArray squeeze(int[] axes)
- Specified by:
squeeze
in interfaceai.djl.ndarray.NDArray
-
unique
public ai.djl.ndarray.NDList unique(java.lang.Integer dim, boolean sorted, boolean returnInverse, boolean returnCounts)
- Specified by:
unique
in interfaceai.djl.ndarray.NDArray
-
logicalAnd
public PtNDArray logicalAnd(ai.djl.ndarray.NDArray other)
- Specified by:
logicalAnd
in interfaceai.djl.ndarray.NDArray
-
logicalOr
public PtNDArray logicalOr(ai.djl.ndarray.NDArray other)
- Specified by:
logicalOr
in interfaceai.djl.ndarray.NDArray
-
logicalXor
public PtNDArray logicalXor(ai.djl.ndarray.NDArray other)
- Specified by:
logicalXor
in interfaceai.djl.ndarray.NDArray
-
logicalNot
public PtNDArray logicalNot()
- Specified by:
logicalNot
in interfaceai.djl.ndarray.NDArray
-
argSort
public PtNDArray argSort(int axis, boolean ascending)
- Specified by:
argSort
in interfaceai.djl.ndarray.NDArray
-
sort
public PtNDArray sort()
- Specified by:
sort
in interfaceai.djl.ndarray.NDArray
-
sort
public PtNDArray sort(int axis)
- Specified by:
sort
in interfaceai.djl.ndarray.NDArray
-
softmax
public PtNDArray softmax(int axis)
- Specified by:
softmax
in interfaceai.djl.ndarray.NDArray
-
logSoftmax
public PtNDArray logSoftmax(int axis)
- Specified by:
logSoftmax
in interfaceai.djl.ndarray.NDArray
-
cumSum
public PtNDArray cumSum()
- Specified by:
cumSum
in interfaceai.djl.ndarray.NDArray
-
cumSum
public PtNDArray cumSum(int axis)
- Specified by:
cumSum
in interfaceai.djl.ndarray.NDArray
-
intern
public void intern(ai.djl.ndarray.NDArray replaced)
- Specified by:
intern
in interfaceai.djl.ndarray.NDArray
-
isInfinite
public PtNDArray isInfinite()
- Specified by:
isInfinite
in interfaceai.djl.ndarray.NDArray
-
isNaN
public PtNDArray isNaN()
- Specified by:
isNaN
in interfaceai.djl.ndarray.NDArray
-
tile
public PtNDArray tile(long repeats)
- Specified by:
tile
in interfaceai.djl.ndarray.NDArray
-
tile
public PtNDArray tile(int axis, long repeats)
- Specified by:
tile
in interfaceai.djl.ndarray.NDArray
-
tile
public PtNDArray tile(long[] repeats)
- Specified by:
tile
in interfaceai.djl.ndarray.NDArray
-
tile
public PtNDArray tile(ai.djl.ndarray.types.Shape desiredShape)
- Specified by:
tile
in interfaceai.djl.ndarray.NDArray
-
repeat
public PtNDArray repeat(long repeats)
- Specified by:
repeat
in interfaceai.djl.ndarray.NDArray
-
repeat
public PtNDArray repeat(int axis, long repeats)
- Specified by:
repeat
in interfaceai.djl.ndarray.NDArray
-
repeat
public PtNDArray repeat(long[] repeats)
- Specified by:
repeat
in interfaceai.djl.ndarray.NDArray
-
repeat
public PtNDArray repeat(ai.djl.ndarray.types.Shape desiredShape)
- Specified by:
repeat
in interfaceai.djl.ndarray.NDArray
-
dot
public PtNDArray dot(ai.djl.ndarray.NDArray other)
- Specified by:
dot
in interfaceai.djl.ndarray.NDArray
-
matMul
public ai.djl.ndarray.NDArray matMul(ai.djl.ndarray.NDArray other)
- Specified by:
matMul
in interfaceai.djl.ndarray.NDArray
-
batchMatMul
public ai.djl.ndarray.NDArray batchMatMul(ai.djl.ndarray.NDArray other)
- Specified by:
batchMatMul
in interfaceai.djl.ndarray.NDArray
-
clip
public PtNDArray clip(java.lang.Number min, java.lang.Number max)
- Specified by:
clip
in interfaceai.djl.ndarray.NDArray
-
swapAxes
public PtNDArray swapAxes(int axis1, int axis2)
- Specified by:
swapAxes
in interfaceai.djl.ndarray.NDArray
-
flip
public ai.djl.ndarray.NDArray flip(int... axes)
- Specified by:
flip
in interfaceai.djl.ndarray.NDArray
-
transpose
public PtNDArray transpose()
- Specified by:
transpose
in interfaceai.djl.ndarray.NDArray
-
transpose
public PtNDArray transpose(int... axes)
- Specified by:
transpose
in interfaceai.djl.ndarray.NDArray
-
broadcast
public PtNDArray broadcast(ai.djl.ndarray.types.Shape shape)
- Specified by:
broadcast
in interfaceai.djl.ndarray.NDArray
-
argMax
public PtNDArray argMax()
- Specified by:
argMax
in interfaceai.djl.ndarray.NDArray
-
argMax
public PtNDArray argMax(int axis)
- Specified by:
argMax
in interfaceai.djl.ndarray.NDArray
-
topK
public ai.djl.ndarray.NDList topK(int k, int axis, boolean largest, boolean sorted)
- Specified by:
topK
in interfaceai.djl.ndarray.NDArray
-
argMin
public PtNDArray argMin()
- Specified by:
argMin
in interfaceai.djl.ndarray.NDArray
-
argMin
public PtNDArray argMin(int axis)
- Specified by:
argMin
in interfaceai.djl.ndarray.NDArray
-
percentile
public PtNDArray percentile(java.lang.Number percentile)
- Specified by:
percentile
in interfaceai.djl.ndarray.NDArray
-
percentile
public PtNDArray percentile(java.lang.Number percentile, int[] axes)
- Specified by:
percentile
in interfaceai.djl.ndarray.NDArray
-
median
public PtNDArray median()
- Specified by:
median
in interfaceai.djl.ndarray.NDArray
-
median
public PtNDArray median(int[] axes)
- Specified by:
median
in interfaceai.djl.ndarray.NDArray
-
toDense
public PtNDArray toDense()
- Specified by:
toDense
in interfaceai.djl.ndarray.NDArray
-
toSparse
public PtNDArray toSparse(ai.djl.ndarray.types.SparseFormat fmt)
- Specified by:
toSparse
in interfaceai.djl.ndarray.NDArray
-
nonzero
public PtNDArray nonzero()
- Specified by:
nonzero
in interfaceai.djl.ndarray.NDArray
-
erfinv
public PtNDArray erfinv()
- Specified by:
erfinv
in interfaceai.djl.ndarray.NDArray
-
inverse
public PtNDArray inverse()
- Specified by:
inverse
in interfaceai.djl.ndarray.NDArray
-
norm
public ai.djl.ndarray.NDArray norm(boolean keepDims)
- Specified by:
norm
in interfaceai.djl.ndarray.NDArray
-
norm
public ai.djl.ndarray.NDArray norm(int order, int[] axes, boolean keepDims)
- Specified by:
norm
in interfaceai.djl.ndarray.NDArray
-
oneHot
public ai.djl.ndarray.NDArray oneHot(int depth)
- Specified by:
oneHot
in interfaceai.djl.ndarray.NDArray
-
oneHot
public ai.djl.ndarray.NDArray oneHot(int depth, ai.djl.ndarray.types.DataType dataType)
- Specified by:
oneHot
in 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:
oneHot
in interfaceai.djl.ndarray.NDArray
-
batchDot
public ai.djl.ndarray.NDArray batchDot(ai.djl.ndarray.NDArray other)
- Specified by:
batchDot
in interfaceai.djl.ndarray.NDArray
-
complex
public ai.djl.ndarray.NDArray complex()
- Specified by:
complex
in interfaceai.djl.ndarray.NDArray
-
real
public ai.djl.ndarray.NDArray real()
- Specified by:
real
in interfaceai.djl.ndarray.NDArray
-
getNDArrayInternal
public PtNDArrayEx getNDArrayInternal()
- Specified by:
getNDArrayInternal
in interfaceai.djl.ndarray.NDArray
-
toString
public java.lang.String toString()
- Overrides:
toString
in classjava.lang.Object
-
equals
public boolean equals(java.lang.Object obj)
- Overrides:
equals
in classjava.lang.Object
-
hashCode
public int hashCode()
- Overrides:
hashCode
in classjava.lang.Object
-
close
public void close()
- Specified by:
close
in interfacejava.lang.AutoCloseable
- Specified by:
close
in interfaceai.djl.ndarray.NDArray
- Specified by:
close
in interfaceai.djl.ndarray.NDResource
- Overrides:
close
in classai.djl.util.NativeResource<java.lang.Long>
-
-