public class PtNDArray
extends ai.djl.util.NativeResource<java.lang.Long>
implements ai.djl.ndarray.NDArray
PtNDArray
is the PyTorch implementation of NDArray
.Constructor and Description |
---|
PtNDArray(PtNDManager manager,
long handle)
Constructs a PyTorch
NDArray from a native handle (internal. |
PtNDArray(PtNDManager manager,
long handle,
java.nio.ByteBuffer data)
Constructs a PyTorch
NDArray from a native handle (internal. |
Modifier and Type | Method and 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) |
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() |
boolean |
contentEquals(ai.djl.ndarray.NDArray other) |
boolean |
contentEquals(java.lang.Number number) |
void |
copyTo(ai.djl.ndarray.NDArray array) |
PtNDArray |
cos() |
PtNDArray |
cosh() |
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) |
PtNDArray |
flatten() |
ai.djl.ndarray.NDArray |
flip(int... axes) |
PtNDArray |
floor() |
ai.djl.ndarray.NDArray |
get(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 |
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) |
ai.djl.ndarray.NDArray |
oneHot(int depth) |
ai.djl.ndarray.NDArray |
oneHot(int depth,
ai.djl.ndarray.types.DataType dataType) |
ai.djl.ndarray.NDArray |
oneHot(int depth,
float onValue,
float offValue,
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) |
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) |
PtNDArray |
rotate90(int times,
int[] axes) |
PtNDArray |
round() |
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 data) |
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 |
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) |
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) |
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() |
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
allClose, allClose, argSort, argSort, booleanMask, broadcast, concat, concat, countNonzero, countNonzero, decode, 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
public PtNDArray(PtNDManager manager, long handle)
NDArray
from a native handle (internal. Use NDManager
instead).manager
- the manager to attach the new array tohandle
- the pointer to the native PyTorch memorypublic PtNDArray(PtNDManager manager, long handle, java.nio.ByteBuffer data)
NDArray
from a native handle (internal. Use NDManager
instead) with the data that is hold on Java side.manager
- the manager to attach the new array tohandle
- the pointer to the native PyTorch memorydata
- the direct buffer of the datapublic PtNDManager getManager()
getManager
in interface ai.djl.ndarray.NDResource
public java.lang.String getName()
getName
in interface ai.djl.ndarray.NDArray
public void setName(java.lang.String name)
setName
in interface ai.djl.ndarray.NDArray
public ai.djl.ndarray.types.DataType getDataType()
getDataType
in interface ai.djl.ndarray.NDArray
public ai.djl.Device getDevice()
getDevice
in interface ai.djl.ndarray.NDArray
public ai.djl.ndarray.types.Shape getShape()
getShape
in interface ai.djl.ndarray.NDArray
public ai.djl.ndarray.types.SparseFormat getSparseFormat()
getSparseFormat
in interface ai.djl.ndarray.NDArray
public PtNDArray toDevice(ai.djl.Device device, boolean copy)
toDevice
in interface ai.djl.ndarray.NDArray
public PtNDArray toType(ai.djl.ndarray.types.DataType dataType, boolean copy)
toType
in interface ai.djl.ndarray.NDArray
public void setRequiresGradient(boolean requiresGrad)
setRequiresGradient
in interface ai.djl.ndarray.NDArray
public PtNDArray getGradient()
getGradient
in interface ai.djl.ndarray.NDArray
public boolean hasGradient()
hasGradient
in interface ai.djl.ndarray.NDArray
public ai.djl.ndarray.NDArray stopGradient()
stopGradient
in interface ai.djl.ndarray.NDArray
public java.nio.ByteBuffer toByteBuffer()
toByteBuffer
in interface ai.djl.ndarray.BytesSupplier
public java.lang.String[] toStringArray(java.nio.charset.Charset charset)
toStringArray
in interface ai.djl.ndarray.NDArray
public void set(java.nio.Buffer data)
set
in interface ai.djl.ndarray.NDArray
public ai.djl.ndarray.NDArray get(long... indices)
get
in interface ai.djl.ndarray.NDArray
public void copyTo(ai.djl.ndarray.NDArray array)
copyTo
in interface ai.djl.ndarray.NDArray
public void attach(ai.djl.ndarray.NDManager manager)
attach
in interface ai.djl.ndarray.NDResource
public void tempAttach(ai.djl.ndarray.NDManager manager)
tempAttach
in interface ai.djl.ndarray.NDResource
public void detach()
detach
in interface ai.djl.ndarray.NDResource
public ai.djl.ndarray.NDArray duplicate()
duplicate
in interface ai.djl.ndarray.NDArray
public PtNDArray booleanMask(ai.djl.ndarray.NDArray index, int axis)
booleanMask
in interface ai.djl.ndarray.NDArray
public ai.djl.ndarray.NDArray sequenceMask(ai.djl.ndarray.NDArray sequenceLength, float value)
sequenceMask
in interface ai.djl.ndarray.NDArray
public ai.djl.ndarray.NDArray sequenceMask(ai.djl.ndarray.NDArray sequenceLength)
sequenceMask
in interface ai.djl.ndarray.NDArray
public boolean contentEquals(java.lang.Number number)
contentEquals
in interface ai.djl.ndarray.NDArray
public boolean contentEquals(ai.djl.ndarray.NDArray other)
contentEquals
in interface ai.djl.ndarray.NDArray
public PtNDArray eq(java.lang.Number n)
eq
in interface ai.djl.ndarray.NDArray
public PtNDArray eq(ai.djl.ndarray.NDArray other)
eq
in interface ai.djl.ndarray.NDArray
public PtNDArray neq(java.lang.Number n)
neq
in interface ai.djl.ndarray.NDArray
public PtNDArray neq(ai.djl.ndarray.NDArray other)
neq
in interface ai.djl.ndarray.NDArray
public PtNDArray gt(java.lang.Number n)
gt
in interface ai.djl.ndarray.NDArray
public PtNDArray gt(ai.djl.ndarray.NDArray other)
gt
in interface ai.djl.ndarray.NDArray
public PtNDArray gte(java.lang.Number n)
gte
in interface ai.djl.ndarray.NDArray
public PtNDArray gte(ai.djl.ndarray.NDArray other)
gte
in interface ai.djl.ndarray.NDArray
public PtNDArray lt(java.lang.Number n)
lt
in interface ai.djl.ndarray.NDArray
public PtNDArray lt(ai.djl.ndarray.NDArray other)
lt
in interface ai.djl.ndarray.NDArray
public PtNDArray lte(java.lang.Number n)
lte
in interface ai.djl.ndarray.NDArray
public PtNDArray lte(ai.djl.ndarray.NDArray other)
lte
in interface ai.djl.ndarray.NDArray
public PtNDArray add(java.lang.Number n)
add
in interface ai.djl.ndarray.NDArray
public PtNDArray add(ai.djl.ndarray.NDArray other)
add
in interface ai.djl.ndarray.NDArray
public PtNDArray sub(java.lang.Number n)
sub
in interface ai.djl.ndarray.NDArray
public PtNDArray sub(ai.djl.ndarray.NDArray other)
sub
in interface ai.djl.ndarray.NDArray
public PtNDArray mul(java.lang.Number n)
mul
in interface ai.djl.ndarray.NDArray
public PtNDArray mul(ai.djl.ndarray.NDArray other)
mul
in interface ai.djl.ndarray.NDArray
public PtNDArray div(java.lang.Number n)
div
in interface ai.djl.ndarray.NDArray
public PtNDArray div(ai.djl.ndarray.NDArray other)
div
in interface ai.djl.ndarray.NDArray
public PtNDArray mod(java.lang.Number n)
mod
in interface ai.djl.ndarray.NDArray
public PtNDArray mod(ai.djl.ndarray.NDArray other)
mod
in interface ai.djl.ndarray.NDArray
public PtNDArray pow(java.lang.Number n)
pow
in interface ai.djl.ndarray.NDArray
public PtNDArray pow(ai.djl.ndarray.NDArray other)
pow
in interface ai.djl.ndarray.NDArray
public PtNDArray addi(java.lang.Number n)
addi
in interface ai.djl.ndarray.NDArray
public PtNDArray addi(ai.djl.ndarray.NDArray other)
addi
in interface ai.djl.ndarray.NDArray
public PtNDArray subi(java.lang.Number n)
subi
in interface ai.djl.ndarray.NDArray
public PtNDArray subi(ai.djl.ndarray.NDArray other)
subi
in interface ai.djl.ndarray.NDArray
public PtNDArray muli(java.lang.Number n)
muli
in interface ai.djl.ndarray.NDArray
public PtNDArray muli(ai.djl.ndarray.NDArray other)
muli
in interface ai.djl.ndarray.NDArray
public PtNDArray divi(java.lang.Number n)
divi
in interface ai.djl.ndarray.NDArray
public PtNDArray divi(ai.djl.ndarray.NDArray other)
divi
in interface ai.djl.ndarray.NDArray
public PtNDArray modi(java.lang.Number n)
modi
in interface ai.djl.ndarray.NDArray
public PtNDArray modi(ai.djl.ndarray.NDArray other)
modi
in interface ai.djl.ndarray.NDArray
public PtNDArray powi(java.lang.Number n)
powi
in interface ai.djl.ndarray.NDArray
public PtNDArray powi(ai.djl.ndarray.NDArray other)
powi
in interface ai.djl.ndarray.NDArray
public PtNDArray sign()
sign
in interface ai.djl.ndarray.NDArray
public PtNDArray signi()
signi
in interface ai.djl.ndarray.NDArray
public PtNDArray maximum(java.lang.Number n)
maximum
in interface ai.djl.ndarray.NDArray
public PtNDArray maximum(ai.djl.ndarray.NDArray other)
maximum
in interface ai.djl.ndarray.NDArray
public PtNDArray minimum(java.lang.Number n)
minimum
in interface ai.djl.ndarray.NDArray
public PtNDArray minimum(ai.djl.ndarray.NDArray other)
minimum
in interface ai.djl.ndarray.NDArray
public PtNDArray all()
all
in interface ai.djl.ndarray.NDArray
public PtNDArray any()
any
in interface ai.djl.ndarray.NDArray
public PtNDArray none()
none
in interface ai.djl.ndarray.NDArray
public PtNDArray neg()
neg
in interface ai.djl.ndarray.NDArray
public PtNDArray negi()
negi
in interface ai.djl.ndarray.NDArray
public PtNDArray abs()
abs
in interface ai.djl.ndarray.NDArray
public PtNDArray square()
square
in interface ai.djl.ndarray.NDArray
public ai.djl.ndarray.NDArray sqrt()
sqrt
in interface ai.djl.ndarray.NDArray
public PtNDArray cbrt()
cbrt
in interface ai.djl.ndarray.NDArray
public PtNDArray floor()
floor
in interface ai.djl.ndarray.NDArray
public PtNDArray ceil()
ceil
in interface ai.djl.ndarray.NDArray
public PtNDArray round()
round
in interface ai.djl.ndarray.NDArray
public PtNDArray trunc()
trunc
in interface ai.djl.ndarray.NDArray
public PtNDArray exp()
exp
in interface ai.djl.ndarray.NDArray
public PtNDArray log()
log
in interface ai.djl.ndarray.NDArray
public PtNDArray log10()
log10
in interface ai.djl.ndarray.NDArray
public PtNDArray log2()
log2
in interface ai.djl.ndarray.NDArray
public PtNDArray sin()
sin
in interface ai.djl.ndarray.NDArray
public PtNDArray cos()
cos
in interface ai.djl.ndarray.NDArray
public PtNDArray tan()
tan
in interface ai.djl.ndarray.NDArray
public PtNDArray asin()
asin
in interface ai.djl.ndarray.NDArray
public PtNDArray acos()
acos
in interface ai.djl.ndarray.NDArray
public PtNDArray atan()
atan
in interface ai.djl.ndarray.NDArray
public PtNDArray sinh()
sinh
in interface ai.djl.ndarray.NDArray
public PtNDArray cosh()
cosh
in interface ai.djl.ndarray.NDArray
public PtNDArray tanh()
tanh
in interface ai.djl.ndarray.NDArray
public PtNDArray asinh()
asinh
in interface ai.djl.ndarray.NDArray
public PtNDArray acosh()
acosh
in interface ai.djl.ndarray.NDArray
public PtNDArray atanh()
atanh
in interface ai.djl.ndarray.NDArray
public PtNDArray toDegrees()
toDegrees
in interface ai.djl.ndarray.NDArray
public PtNDArray toRadians()
toRadians
in interface ai.djl.ndarray.NDArray
public PtNDArray max()
max
in interface ai.djl.ndarray.NDArray
public PtNDArray max(int[] axes, boolean keepDims)
max
in interface ai.djl.ndarray.NDArray
public PtNDArray min()
min
in interface ai.djl.ndarray.NDArray
public PtNDArray min(int[] axes, boolean keepDims)
min
in interface ai.djl.ndarray.NDArray
public PtNDArray sum()
sum
in interface ai.djl.ndarray.NDArray
public PtNDArray sum(int[] axes, boolean keepDims)
sum
in interface ai.djl.ndarray.NDArray
public PtNDArray prod()
prod
in interface ai.djl.ndarray.NDArray
public PtNDArray prod(int[] axes, boolean keepDims)
prod
in interface ai.djl.ndarray.NDArray
public PtNDArray mean()
mean
in interface ai.djl.ndarray.NDArray
public PtNDArray mean(int[] axes, boolean keepDims)
mean
in interface ai.djl.ndarray.NDArray
public PtNDArray rotate90(int times, int[] axes)
rotate90
in interface ai.djl.ndarray.NDArray
public PtNDArray trace(int offset, int axis1, int axis2)
trace
in interface ai.djl.ndarray.NDArray
public ai.djl.ndarray.NDList split(long sections, int axis)
split
in interface ai.djl.ndarray.NDArray
public ai.djl.ndarray.NDList split(long[] indices, int axis)
split
in interface ai.djl.ndarray.NDArray
public PtNDArray flatten()
flatten
in interface ai.djl.ndarray.NDArray
public PtNDArray reshape(ai.djl.ndarray.types.Shape shape)
reshape
in interface ai.djl.ndarray.NDArray
public PtNDArray expandDims(int axis)
expandDims
in interface ai.djl.ndarray.NDArray
public PtNDArray squeeze()
squeeze
in interface ai.djl.ndarray.NDArray
public PtNDArray squeeze(int axis)
squeeze
in interface ai.djl.ndarray.NDArray
public PtNDArray squeeze(int[] axes)
squeeze
in interface ai.djl.ndarray.NDArray
public PtNDArray logicalAnd(ai.djl.ndarray.NDArray other)
logicalAnd
in interface ai.djl.ndarray.NDArray
public PtNDArray logicalOr(ai.djl.ndarray.NDArray other)
logicalOr
in interface ai.djl.ndarray.NDArray
public PtNDArray logicalXor(ai.djl.ndarray.NDArray other)
logicalXor
in interface ai.djl.ndarray.NDArray
public PtNDArray logicalNot()
logicalNot
in interface ai.djl.ndarray.NDArray
public PtNDArray argSort(int axis, boolean ascending)
argSort
in interface ai.djl.ndarray.NDArray
public PtNDArray sort()
sort
in interface ai.djl.ndarray.NDArray
public PtNDArray sort(int axis)
sort
in interface ai.djl.ndarray.NDArray
public PtNDArray softmax(int axis)
softmax
in interface ai.djl.ndarray.NDArray
public PtNDArray logSoftmax(int axis)
logSoftmax
in interface ai.djl.ndarray.NDArray
public PtNDArray cumSum()
cumSum
in interface ai.djl.ndarray.NDArray
public PtNDArray cumSum(int axis)
cumSum
in interface ai.djl.ndarray.NDArray
public void intern(ai.djl.ndarray.NDArray replaced)
intern
in interface ai.djl.ndarray.NDArray
public PtNDArray isInfinite()
isInfinite
in interface ai.djl.ndarray.NDArray
public PtNDArray isNaN()
isNaN
in interface ai.djl.ndarray.NDArray
public PtNDArray tile(long repeats)
tile
in interface ai.djl.ndarray.NDArray
public PtNDArray tile(int axis, long repeats)
tile
in interface ai.djl.ndarray.NDArray
public PtNDArray tile(long[] repeats)
tile
in interface ai.djl.ndarray.NDArray
public PtNDArray tile(ai.djl.ndarray.types.Shape desiredShape)
tile
in interface ai.djl.ndarray.NDArray
public PtNDArray repeat(long repeats)
repeat
in interface ai.djl.ndarray.NDArray
public PtNDArray repeat(int axis, long repeats)
repeat
in interface ai.djl.ndarray.NDArray
public PtNDArray repeat(long[] repeats)
repeat
in interface ai.djl.ndarray.NDArray
public PtNDArray repeat(ai.djl.ndarray.types.Shape desiredShape)
repeat
in interface ai.djl.ndarray.NDArray
public PtNDArray dot(ai.djl.ndarray.NDArray other)
dot
in interface ai.djl.ndarray.NDArray
public ai.djl.ndarray.NDArray matMul(ai.djl.ndarray.NDArray other)
matMul
in interface ai.djl.ndarray.NDArray
public PtNDArray clip(java.lang.Number min, java.lang.Number max)
clip
in interface ai.djl.ndarray.NDArray
public PtNDArray swapAxes(int axis1, int axis2)
swapAxes
in interface ai.djl.ndarray.NDArray
public ai.djl.ndarray.NDArray flip(int... axes)
flip
in interface ai.djl.ndarray.NDArray
public PtNDArray transpose()
transpose
in interface ai.djl.ndarray.NDArray
public PtNDArray transpose(int... axes)
transpose
in interface ai.djl.ndarray.NDArray
public PtNDArray broadcast(ai.djl.ndarray.types.Shape shape)
broadcast
in interface ai.djl.ndarray.NDArray
public PtNDArray argMax()
argMax
in interface ai.djl.ndarray.NDArray
public PtNDArray argMax(int axis)
argMax
in interface ai.djl.ndarray.NDArray
public PtNDArray argMin()
argMin
in interface ai.djl.ndarray.NDArray
public PtNDArray argMin(int axis)
argMin
in interface ai.djl.ndarray.NDArray
public PtNDArray percentile(java.lang.Number percentile)
percentile
in interface ai.djl.ndarray.NDArray
public PtNDArray percentile(java.lang.Number percentile, int[] axes)
percentile
in interface ai.djl.ndarray.NDArray
public PtNDArray median()
median
in interface ai.djl.ndarray.NDArray
public PtNDArray median(int[] axes)
median
in interface ai.djl.ndarray.NDArray
public PtNDArray toDense()
toDense
in interface ai.djl.ndarray.NDArray
public PtNDArray toSparse(ai.djl.ndarray.types.SparseFormat fmt)
toSparse
in interface ai.djl.ndarray.NDArray
public PtNDArray nonzero()
nonzero
in interface ai.djl.ndarray.NDArray
public PtNDArray erfinv()
erfinv
in interface ai.djl.ndarray.NDArray
public ai.djl.ndarray.NDArray norm(boolean keepDims)
norm
in interface ai.djl.ndarray.NDArray
public ai.djl.ndarray.NDArray norm(int order, int[] axes, boolean keepDims)
norm
in interface ai.djl.ndarray.NDArray
public ai.djl.ndarray.NDArray oneHot(int depth)
oneHot
in interface ai.djl.ndarray.NDArray
public ai.djl.ndarray.NDArray oneHot(int depth, ai.djl.ndarray.types.DataType dataType)
oneHot
in interface ai.djl.ndarray.NDArray
public ai.djl.ndarray.NDArray oneHot(int depth, float onValue, float offValue, ai.djl.ndarray.types.DataType dataType)
oneHot
in interface ai.djl.ndarray.NDArray
public ai.djl.ndarray.NDArray batchDot(ai.djl.ndarray.NDArray other)
batchDot
in interface ai.djl.ndarray.NDArray
public PtNDArrayEx getNDArrayInternal()
getNDArrayInternal
in interface ai.djl.ndarray.NDArray
public java.lang.String toString()
toString
in class java.lang.Object
public boolean equals(java.lang.Object obj)
equals
in class java.lang.Object
public int hashCode()
hashCode
in class java.lang.Object
public void close()
close
in interface ai.djl.ndarray.NDArray
close
in interface ai.djl.ndarray.NDResource
close
in interface java.lang.AutoCloseable
close
in class ai.djl.util.NativeResource<java.lang.Long>