Package ai.djl.tensorflow.engine
Class TfNDArray
- java.lang.Object
-
- ai.djl.util.NativeResource<org.tensorflow.internal.c_api.TFE_TensorHandle>
-
- ai.djl.tensorflow.engine.TfNDArray
-
- All Implemented Interfaces:
ai.djl.ndarray.BytesSupplier
,ai.djl.ndarray.NDArray
,ai.djl.ndarray.NDResource
,java.lang.AutoCloseable
public class TfNDArray extends ai.djl.util.NativeResource<org.tensorflow.internal.c_api.TFE_TensorHandle> implements ai.djl.ndarray.NDArray
TfNDArray
is the TensorFlow implementation ofNDArray
.
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description ai.djl.ndarray.NDArray
abs()
ai.djl.ndarray.NDArray
acos()
ai.djl.ndarray.NDArray
acosh()
ai.djl.ndarray.NDArray
add(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
add(java.lang.Number n)
ai.djl.ndarray.NDArray
addi(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
addi(java.lang.Number n)
ai.djl.ndarray.NDArray
all()
ai.djl.ndarray.NDArray
any()
ai.djl.ndarray.NDArray
argMax()
ai.djl.ndarray.NDArray
argMax(int axis)
ai.djl.ndarray.NDArray
argMin()
ai.djl.ndarray.NDArray
argMin(int axis)
ai.djl.ndarray.NDArray
argSort(int axis, boolean ascending)
ai.djl.ndarray.NDArray
asin()
ai.djl.ndarray.NDArray
asinh()
ai.djl.ndarray.NDArray
atan()
ai.djl.ndarray.NDArray
atanh()
void
attach(ai.djl.ndarray.NDManager manager)
ai.djl.ndarray.NDArray
batchDot(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
booleanMask(ai.djl.ndarray.NDArray index, int axis)
ai.djl.ndarray.NDArray
broadcast(ai.djl.ndarray.types.Shape shape)
ai.djl.ndarray.NDArray
cbrt()
ai.djl.ndarray.NDArray
ceil()
ai.djl.ndarray.NDArray
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)
void
copyTo(ai.djl.ndarray.NDArray ndArray)
ai.djl.ndarray.NDArray
cos()
ai.djl.ndarray.NDArray
cosh()
ai.djl.ndarray.NDArray
cumProd(int axis)
ai.djl.ndarray.NDArray
cumProd(int axis, ai.djl.ndarray.types.DataType dataType)
ai.djl.ndarray.NDArray
cumSum()
ai.djl.ndarray.NDArray
cumSum(int axis)
void
detach()
ai.djl.ndarray.NDArray
div(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
div(java.lang.Number n)
ai.djl.ndarray.NDArray
divi(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
divi(java.lang.Number n)
ai.djl.ndarray.NDArray
dot(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
duplicate()
ai.djl.ndarray.NDArray
eq(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
eq(java.lang.Number n)
boolean
equals(java.lang.Object obj)
ai.djl.ndarray.NDArray
erfinv()
ai.djl.ndarray.NDArray
exp()
ai.djl.ndarray.NDArray
expandDims(int axis)
ai.djl.ndarray.NDArray
fft(long length, long axis)
ai.djl.ndarray.NDArray
flatten()
ai.djl.ndarray.NDArray
flatten(int startDim, int endDim)
ai.djl.ndarray.NDArray
flip(int... axes)
ai.djl.ndarray.NDArray
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.types.DataType
getDataType()
ai.djl.Device
getDevice()
ai.djl.ndarray.NDArray
getGradient()
TfNDManager
getManager()
java.lang.String
getName()
ai.djl.ndarray.internal.NDArrayEx
getNDArrayInternal()
ai.djl.ndarray.types.Shape
getShape()
ai.djl.ndarray.types.SparseFormat
getSparseFormat()
ai.djl.ndarray.NDArray
gt(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
gt(java.lang.Number n)
ai.djl.ndarray.NDArray
gte(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
gte(java.lang.Number n)
boolean
hasGradient()
int
hashCode()
void
intern(ai.djl.ndarray.NDArray replaced)
ai.djl.ndarray.NDArray
inverse()
ai.djl.ndarray.NDArray
isInfinite()
ai.djl.ndarray.NDArray
isNaN()
ai.djl.ndarray.NDArray
log()
ai.djl.ndarray.NDArray
log10()
ai.djl.ndarray.NDArray
log2()
ai.djl.ndarray.NDArray
logicalAnd(ai.djl.ndarray.NDArray n)
ai.djl.ndarray.NDArray
logicalNot()
ai.djl.ndarray.NDArray
logicalOr(ai.djl.ndarray.NDArray n)
ai.djl.ndarray.NDArray
logicalXor(ai.djl.ndarray.NDArray n)
ai.djl.ndarray.NDArray
logSoftmax(int axis)
ai.djl.ndarray.NDArray
lt(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
lt(java.lang.Number n)
ai.djl.ndarray.NDArray
lte(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
lte(java.lang.Number n)
ai.djl.ndarray.NDArray
matMul(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
max()
ai.djl.ndarray.NDArray
max(int[] axes, boolean keepDims)
ai.djl.ndarray.NDArray
maximum(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
maximum(java.lang.Number n)
ai.djl.ndarray.NDArray
mean()
ai.djl.ndarray.NDArray
mean(int[] axes, boolean keepDims)
ai.djl.ndarray.NDArray
median()
ai.djl.ndarray.NDArray
median(int[] axes)
ai.djl.ndarray.NDArray
min()
ai.djl.ndarray.NDArray
min(int[] axes, boolean keepDims)
ai.djl.ndarray.NDArray
minimum(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
minimum(java.lang.Number n)
ai.djl.ndarray.NDArray
mod(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
mod(java.lang.Number n)
ai.djl.ndarray.NDArray
modi(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
modi(java.lang.Number n)
ai.djl.ndarray.NDArray
mul(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
mul(java.lang.Number n)
ai.djl.ndarray.NDArray
muli(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
muli(java.lang.Number n)
ai.djl.ndarray.NDArray
neg()
ai.djl.ndarray.NDArray
negi()
ai.djl.ndarray.NDArray
neq(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
neq(java.lang.Number n)
ai.djl.ndarray.NDArray
nonzero()
ai.djl.ndarray.NDArray
norm(boolean keepDims)
ai.djl.ndarray.NDArray
norm(int ord, int[] axes, boolean keepDims)
ai.djl.ndarray.NDArray
normalize(double p, long dim, double eps)
ai.djl.ndarray.NDArray
oneHot(int depth, float onValue, float offValue, ai.djl.ndarray.types.DataType dataType)
ai.djl.ndarray.NDArray
percentile(java.lang.Number percentile)
ai.djl.ndarray.NDArray
percentile(java.lang.Number percentile, int[] dimension)
ai.djl.ndarray.NDArray
pow(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
pow(java.lang.Number n)
ai.djl.ndarray.NDArray
powi(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
powi(java.lang.Number n)
ai.djl.ndarray.NDArray
prod()
ai.djl.ndarray.NDArray
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()
ai.djl.ndarray.NDArray
repeat(int axis, long repeats)
ai.djl.ndarray.NDArray
repeat(long repeats)
ai.djl.ndarray.NDArray
repeat(long[] repeats)
ai.djl.ndarray.NDArray
repeat(ai.djl.ndarray.types.Shape desiredShape)
ai.djl.ndarray.NDArray
reshape(ai.djl.ndarray.types.Shape shape)
void
returnResource(ai.djl.ndarray.NDManager manager)
ai.djl.ndarray.NDArray
rotate90(int times, int[] axes)
ai.djl.ndarray.NDArray
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)
protected void
setHandle(org.tensorflow.internal.c_api.TFE_TensorHandle newHandle)
void
setName(java.lang.String name)
void
setRequiresGradient(boolean requiresGrad)
ai.djl.ndarray.NDArray
sign()
ai.djl.ndarray.NDArray
signi()
ai.djl.ndarray.NDArray
sin()
ai.djl.ndarray.NDArray
sinh()
ai.djl.ndarray.NDArray
softmax(int axis)
ai.djl.ndarray.NDArray
sort()
ai.djl.ndarray.NDArray
sort(int axis)
ai.djl.ndarray.NDList
split(long[] indices, int axis)
ai.djl.ndarray.NDArray
sqrt()
ai.djl.ndarray.NDArray
square()
ai.djl.ndarray.NDArray
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()
ai.djl.ndarray.NDArray
sub(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
sub(java.lang.Number n)
ai.djl.ndarray.NDArray
subi(ai.djl.ndarray.NDArray other)
ai.djl.ndarray.NDArray
subi(java.lang.Number n)
ai.djl.ndarray.NDArray
sum()
ai.djl.ndarray.NDArray
sum(int[] axes, boolean keepDims)
ai.djl.ndarray.NDArray
take(ai.djl.ndarray.NDManager manager, ai.djl.ndarray.NDArray index)
ai.djl.ndarray.NDArray
tan()
ai.djl.ndarray.NDArray
tanh()
void
tempAttach(ai.djl.ndarray.NDManager manager)
ai.djl.ndarray.NDArray
tile(int axis, long repeats)
ai.djl.ndarray.NDArray
tile(long repeats)
ai.djl.ndarray.NDArray
tile(long[] repeats)
ai.djl.ndarray.NDArray
tile(ai.djl.ndarray.types.Shape desiredShape)
java.nio.ByteBuffer
toByteBuffer()
ai.djl.ndarray.NDArray
toDegrees()
ai.djl.ndarray.NDArray
toDense()
ai.djl.ndarray.NDArray
toDevice(ai.djl.Device device, boolean copy)
ai.djl.ndarray.NDArray
toRadians()
ai.djl.ndarray.NDArray
toSparse(ai.djl.ndarray.types.SparseFormat fmt)
java.lang.String
toString()
java.lang.String[]
toStringArray(java.nio.charset.Charset charset)
ai.djl.ndarray.NDArray
toType(ai.djl.ndarray.types.DataType dataType, boolean copy)
ai.djl.ndarray.NDArray
trace(int offset, int axis1, int axis2)
ai.djl.ndarray.NDArray
transpose()
ai.djl.ndarray.NDArray
transpose(int... dimensions)
ai.djl.ndarray.NDArray
trunc()
-
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, fft, get, get, get, get, get, get, getBoolean, getByte, getDouble, getFloat, getInt, getLong, getResourceNDArrays, getScalar, getUid, getUint8, isEmpty, isReleased, isScalar, isSparse, like, max, mean, min, none, norm, norm, norm, normalize, normalize, oneHot, oneHot, onesLike, prod, reshape, scaleGradient, set, set, set, set, set, set, set, set, set, setScalar, shapeEquals, size, size, split, split, split, squeeze, squeeze, stack, stack, stft, sum, swapAxes, take, toArray, toBooleanArray, toByteArray, toDebugString, toDebugString, toDebugString, toDoubleArray, toFloatArray, toIntArray, toLongArray, toStringArray, toUint8Array, trace, trace, zerosLike
-
-
-
-
Method Detail
-
getManager
public TfNDManager 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 ai.djl.ndarray.NDArray toDevice(ai.djl.Device device, boolean copy)
- Specified by:
toDevice
in interfaceai.djl.ndarray.NDArray
-
toType
public ai.djl.ndarray.NDArray 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 ai.djl.ndarray.NDArray 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
-
toStringArray
public java.lang.String[] toStringArray(java.nio.charset.Charset charset)
- Specified by:
toStringArray
in interfaceai.djl.ndarray.NDArray
-
toByteBuffer
public java.nio.ByteBuffer toByteBuffer()
- Specified by:
toByteBuffer
in interfaceai.djl.ndarray.BytesSupplier
-
set
public void set(java.nio.Buffer buffer)
- Specified by:
set
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
-
copyTo
public void copyTo(ai.djl.ndarray.NDArray ndArray)
- Specified by:
copyTo
in interfaceai.djl.ndarray.NDArray
-
duplicate
public ai.djl.ndarray.NDArray duplicate()
- Specified by:
duplicate
in interfaceai.djl.ndarray.NDArray
-
booleanMask
public ai.djl.ndarray.NDArray 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 ai.djl.ndarray.NDArray eq(java.lang.Number n)
- Specified by:
eq
in interfaceai.djl.ndarray.NDArray
-
eq
public ai.djl.ndarray.NDArray eq(ai.djl.ndarray.NDArray other)
- Specified by:
eq
in interfaceai.djl.ndarray.NDArray
-
neq
public ai.djl.ndarray.NDArray neq(java.lang.Number n)
- Specified by:
neq
in interfaceai.djl.ndarray.NDArray
-
neq
public ai.djl.ndarray.NDArray neq(ai.djl.ndarray.NDArray other)
- Specified by:
neq
in interfaceai.djl.ndarray.NDArray
-
gt
public ai.djl.ndarray.NDArray gt(java.lang.Number n)
- Specified by:
gt
in interfaceai.djl.ndarray.NDArray
-
gt
public ai.djl.ndarray.NDArray gt(ai.djl.ndarray.NDArray other)
- Specified by:
gt
in interfaceai.djl.ndarray.NDArray
-
gte
public ai.djl.ndarray.NDArray gte(java.lang.Number n)
- Specified by:
gte
in interfaceai.djl.ndarray.NDArray
-
gte
public ai.djl.ndarray.NDArray gte(ai.djl.ndarray.NDArray other)
- Specified by:
gte
in interfaceai.djl.ndarray.NDArray
-
lt
public ai.djl.ndarray.NDArray lt(java.lang.Number n)
- Specified by:
lt
in interfaceai.djl.ndarray.NDArray
-
lt
public ai.djl.ndarray.NDArray lt(ai.djl.ndarray.NDArray other)
- Specified by:
lt
in interfaceai.djl.ndarray.NDArray
-
lte
public ai.djl.ndarray.NDArray lte(java.lang.Number n)
- Specified by:
lte
in interfaceai.djl.ndarray.NDArray
-
lte
public ai.djl.ndarray.NDArray lte(ai.djl.ndarray.NDArray other)
- Specified by:
lte
in interfaceai.djl.ndarray.NDArray
-
all
public ai.djl.ndarray.NDArray all()
- Specified by:
all
in interfaceai.djl.ndarray.NDArray
-
any
public ai.djl.ndarray.NDArray any()
- Specified by:
any
in interfaceai.djl.ndarray.NDArray
-
erfinv
public ai.djl.ndarray.NDArray erfinv()
- Specified by:
erfinv
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 ord, int[] axes, boolean keepDims)
- Specified by:
norm
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
-
add
public ai.djl.ndarray.NDArray add(java.lang.Number n)
- Specified by:
add
in interfaceai.djl.ndarray.NDArray
-
add
public ai.djl.ndarray.NDArray add(ai.djl.ndarray.NDArray other)
- Specified by:
add
in interfaceai.djl.ndarray.NDArray
-
sub
public ai.djl.ndarray.NDArray sub(java.lang.Number n)
- Specified by:
sub
in interfaceai.djl.ndarray.NDArray
-
sub
public ai.djl.ndarray.NDArray sub(ai.djl.ndarray.NDArray other)
- Specified by:
sub
in interfaceai.djl.ndarray.NDArray
-
mul
public ai.djl.ndarray.NDArray mul(java.lang.Number n)
- Specified by:
mul
in interfaceai.djl.ndarray.NDArray
-
mul
public ai.djl.ndarray.NDArray mul(ai.djl.ndarray.NDArray other)
- Specified by:
mul
in interfaceai.djl.ndarray.NDArray
-
div
public ai.djl.ndarray.NDArray div(java.lang.Number n)
- Specified by:
div
in interfaceai.djl.ndarray.NDArray
-
div
public ai.djl.ndarray.NDArray div(ai.djl.ndarray.NDArray other)
- Specified by:
div
in interfaceai.djl.ndarray.NDArray
-
mod
public ai.djl.ndarray.NDArray mod(java.lang.Number n)
- Specified by:
mod
in interfaceai.djl.ndarray.NDArray
-
mod
public ai.djl.ndarray.NDArray mod(ai.djl.ndarray.NDArray other)
- Specified by:
mod
in interfaceai.djl.ndarray.NDArray
-
pow
public ai.djl.ndarray.NDArray pow(java.lang.Number n)
- Specified by:
pow
in interfaceai.djl.ndarray.NDArray
-
pow
public ai.djl.ndarray.NDArray pow(ai.djl.ndarray.NDArray other)
- Specified by:
pow
in interfaceai.djl.ndarray.NDArray
-
maximum
public ai.djl.ndarray.NDArray maximum(java.lang.Number n)
- Specified by:
maximum
in interfaceai.djl.ndarray.NDArray
-
maximum
public ai.djl.ndarray.NDArray maximum(ai.djl.ndarray.NDArray other)
- Specified by:
maximum
in interfaceai.djl.ndarray.NDArray
-
minimum
public ai.djl.ndarray.NDArray minimum(java.lang.Number n)
- Specified by:
minimum
in interfaceai.djl.ndarray.NDArray
-
minimum
public ai.djl.ndarray.NDArray minimum(ai.djl.ndarray.NDArray other)
- Specified by:
minimum
in interfaceai.djl.ndarray.NDArray
-
addi
public ai.djl.ndarray.NDArray addi(java.lang.Number n)
- Specified by:
addi
in interfaceai.djl.ndarray.NDArray
-
addi
public ai.djl.ndarray.NDArray addi(ai.djl.ndarray.NDArray other)
- Specified by:
addi
in interfaceai.djl.ndarray.NDArray
-
subi
public ai.djl.ndarray.NDArray subi(java.lang.Number n)
- Specified by:
subi
in interfaceai.djl.ndarray.NDArray
-
subi
public ai.djl.ndarray.NDArray subi(ai.djl.ndarray.NDArray other)
- Specified by:
subi
in interfaceai.djl.ndarray.NDArray
-
muli
public ai.djl.ndarray.NDArray muli(java.lang.Number n)
- Specified by:
muli
in interfaceai.djl.ndarray.NDArray
-
muli
public ai.djl.ndarray.NDArray muli(ai.djl.ndarray.NDArray other)
- Specified by:
muli
in interfaceai.djl.ndarray.NDArray
-
divi
public ai.djl.ndarray.NDArray divi(java.lang.Number n)
- Specified by:
divi
in interfaceai.djl.ndarray.NDArray
-
divi
public ai.djl.ndarray.NDArray divi(ai.djl.ndarray.NDArray other)
- Specified by:
divi
in interfaceai.djl.ndarray.NDArray
-
toSparse
public ai.djl.ndarray.NDArray toSparse(ai.djl.ndarray.types.SparseFormat fmt)
- Specified by:
toSparse
in interfaceai.djl.ndarray.NDArray
-
modi
public ai.djl.ndarray.NDArray modi(java.lang.Number n)
- Specified by:
modi
in interfaceai.djl.ndarray.NDArray
-
modi
public ai.djl.ndarray.NDArray modi(ai.djl.ndarray.NDArray other)
- Specified by:
modi
in interfaceai.djl.ndarray.NDArray
-
powi
public ai.djl.ndarray.NDArray powi(java.lang.Number n)
- Specified by:
powi
in interfaceai.djl.ndarray.NDArray
-
powi
public ai.djl.ndarray.NDArray powi(ai.djl.ndarray.NDArray other)
- Specified by:
powi
in interfaceai.djl.ndarray.NDArray
-
sign
public ai.djl.ndarray.NDArray sign()
- Specified by:
sign
in interfaceai.djl.ndarray.NDArray
-
signi
public ai.djl.ndarray.NDArray signi()
- Specified by:
signi
in interfaceai.djl.ndarray.NDArray
-
neg
public ai.djl.ndarray.NDArray neg()
- Specified by:
neg
in interfaceai.djl.ndarray.NDArray
-
negi
public ai.djl.ndarray.NDArray negi()
- Specified by:
negi
in interfaceai.djl.ndarray.NDArray
-
abs
public ai.djl.ndarray.NDArray abs()
- Specified by:
abs
in interfaceai.djl.ndarray.NDArray
-
square
public ai.djl.ndarray.NDArray 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 ai.djl.ndarray.NDArray cbrt()
- Specified by:
cbrt
in interfaceai.djl.ndarray.NDArray
-
floor
public ai.djl.ndarray.NDArray floor()
- Specified by:
floor
in interfaceai.djl.ndarray.NDArray
-
ceil
public ai.djl.ndarray.NDArray ceil()
- Specified by:
ceil
in interfaceai.djl.ndarray.NDArray
-
round
public ai.djl.ndarray.NDArray round()
- Specified by:
round
in interfaceai.djl.ndarray.NDArray
-
trunc
public ai.djl.ndarray.NDArray trunc()
- Specified by:
trunc
in interfaceai.djl.ndarray.NDArray
-
exp
public ai.djl.ndarray.NDArray 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 ai.djl.ndarray.NDArray log()
- Specified by:
log
in interfaceai.djl.ndarray.NDArray
-
log10
public ai.djl.ndarray.NDArray log10()
- Specified by:
log10
in interfaceai.djl.ndarray.NDArray
-
log2
public ai.djl.ndarray.NDArray log2()
- Specified by:
log2
in interfaceai.djl.ndarray.NDArray
-
sin
public ai.djl.ndarray.NDArray sin()
- Specified by:
sin
in interfaceai.djl.ndarray.NDArray
-
cos
public ai.djl.ndarray.NDArray cos()
- Specified by:
cos
in interfaceai.djl.ndarray.NDArray
-
tan
public ai.djl.ndarray.NDArray tan()
- Specified by:
tan
in interfaceai.djl.ndarray.NDArray
-
asin
public ai.djl.ndarray.NDArray asin()
- Specified by:
asin
in interfaceai.djl.ndarray.NDArray
-
acos
public ai.djl.ndarray.NDArray acos()
- Specified by:
acos
in interfaceai.djl.ndarray.NDArray
-
atan
public ai.djl.ndarray.NDArray atan()
- Specified by:
atan
in interfaceai.djl.ndarray.NDArray
-
sinh
public ai.djl.ndarray.NDArray sinh()
- Specified by:
sinh
in interfaceai.djl.ndarray.NDArray
-
cosh
public ai.djl.ndarray.NDArray cosh()
- Specified by:
cosh
in interfaceai.djl.ndarray.NDArray
-
tanh
public ai.djl.ndarray.NDArray tanh()
- Specified by:
tanh
in interfaceai.djl.ndarray.NDArray
-
asinh
public ai.djl.ndarray.NDArray asinh()
- Specified by:
asinh
in interfaceai.djl.ndarray.NDArray
-
acosh
public ai.djl.ndarray.NDArray acosh()
- Specified by:
acosh
in interfaceai.djl.ndarray.NDArray
-
atanh
public ai.djl.ndarray.NDArray atanh()
- Specified by:
atanh
in interfaceai.djl.ndarray.NDArray
-
toDegrees
public ai.djl.ndarray.NDArray toDegrees()
- Specified by:
toDegrees
in interfaceai.djl.ndarray.NDArray
-
toRadians
public ai.djl.ndarray.NDArray toRadians()
- Specified by:
toRadians
in interfaceai.djl.ndarray.NDArray
-
max
public ai.djl.ndarray.NDArray max()
- Specified by:
max
in interfaceai.djl.ndarray.NDArray
-
max
public ai.djl.ndarray.NDArray max(int[] axes, boolean keepDims)
- Specified by:
max
in interfaceai.djl.ndarray.NDArray
-
min
public ai.djl.ndarray.NDArray min()
- Specified by:
min
in interfaceai.djl.ndarray.NDArray
-
min
public ai.djl.ndarray.NDArray min(int[] axes, boolean keepDims)
- Specified by:
min
in interfaceai.djl.ndarray.NDArray
-
sum
public ai.djl.ndarray.NDArray sum()
- Specified by:
sum
in interfaceai.djl.ndarray.NDArray
-
sum
public ai.djl.ndarray.NDArray 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 ai.djl.ndarray.NDArray prod()
- Specified by:
prod
in interfaceai.djl.ndarray.NDArray
-
prod
public ai.djl.ndarray.NDArray prod(int[] axes, boolean keepDims)
- Specified by:
prod
in interfaceai.djl.ndarray.NDArray
-
mean
public ai.djl.ndarray.NDArray mean()
- Specified by:
mean
in interfaceai.djl.ndarray.NDArray
-
mean
public ai.djl.ndarray.NDArray mean(int[] axes, boolean keepDims)
- Specified by:
mean
in interfaceai.djl.ndarray.NDArray
-
normalize
public ai.djl.ndarray.NDArray normalize(double p, long dim, double eps)
- Specified by:
normalize
in interfaceai.djl.ndarray.NDArray
-
rotate90
public ai.djl.ndarray.NDArray rotate90(int times, int[] axes)
- Specified by:
rotate90
in interfaceai.djl.ndarray.NDArray
-
trace
public ai.djl.ndarray.NDArray trace(int offset, int axis1, int axis2)
- Specified by:
trace
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 ai.djl.ndarray.NDArray 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 ai.djl.ndarray.NDArray reshape(ai.djl.ndarray.types.Shape shape)
- Specified by:
reshape
in interfaceai.djl.ndarray.NDArray
-
expandDims
public ai.djl.ndarray.NDArray expandDims(int axis)
- Specified by:
expandDims
in interfaceai.djl.ndarray.NDArray
-
squeeze
public ai.djl.ndarray.NDArray squeeze(int[] axes)
- Specified by:
squeeze
in interfaceai.djl.ndarray.NDArray
-
logicalAnd
public ai.djl.ndarray.NDArray logicalAnd(ai.djl.ndarray.NDArray n)
- Specified by:
logicalAnd
in interfaceai.djl.ndarray.NDArray
-
logicalOr
public ai.djl.ndarray.NDArray logicalOr(ai.djl.ndarray.NDArray n)
- Specified by:
logicalOr
in interfaceai.djl.ndarray.NDArray
-
logicalXor
public ai.djl.ndarray.NDArray logicalXor(ai.djl.ndarray.NDArray n)
- Specified by:
logicalXor
in interfaceai.djl.ndarray.NDArray
-
logicalNot
public ai.djl.ndarray.NDArray logicalNot()
- Specified by:
logicalNot
in interfaceai.djl.ndarray.NDArray
-
argSort
public ai.djl.ndarray.NDArray argSort(int axis, boolean ascending)
- Specified by:
argSort
in interfaceai.djl.ndarray.NDArray
-
sort
public ai.djl.ndarray.NDArray sort(int axis)
- Specified by:
sort
in interfaceai.djl.ndarray.NDArray
-
sort
public ai.djl.ndarray.NDArray sort()
- Specified by:
sort
in interfaceai.djl.ndarray.NDArray
-
softmax
public ai.djl.ndarray.NDArray softmax(int axis)
- Specified by:
softmax
in interfaceai.djl.ndarray.NDArray
-
logSoftmax
public ai.djl.ndarray.NDArray logSoftmax(int axis)
- Specified by:
logSoftmax
in interfaceai.djl.ndarray.NDArray
-
cumSum
public ai.djl.ndarray.NDArray cumSum(int axis)
- Specified by:
cumSum
in interfaceai.djl.ndarray.NDArray
-
cumSum
public ai.djl.ndarray.NDArray cumSum()
- 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 ai.djl.ndarray.NDArray isInfinite()
- Specified by:
isInfinite
in interfaceai.djl.ndarray.NDArray
-
inverse
public ai.djl.ndarray.NDArray inverse()
- Specified by:
inverse
in interfaceai.djl.ndarray.NDArray
-
isNaN
public ai.djl.ndarray.NDArray isNaN()
- Specified by:
isNaN
in interfaceai.djl.ndarray.NDArray
-
tile
public ai.djl.ndarray.NDArray tile(long repeats)
- Specified by:
tile
in interfaceai.djl.ndarray.NDArray
-
tile
public ai.djl.ndarray.NDArray tile(int axis, long repeats)
- Specified by:
tile
in interfaceai.djl.ndarray.NDArray
-
tile
public ai.djl.ndarray.NDArray tile(long[] repeats)
- Specified by:
tile
in interfaceai.djl.ndarray.NDArray
-
tile
public ai.djl.ndarray.NDArray tile(ai.djl.ndarray.types.Shape desiredShape)
- Specified by:
tile
in interfaceai.djl.ndarray.NDArray
-
repeat
public ai.djl.ndarray.NDArray repeat(long repeats)
- Specified by:
repeat
in interfaceai.djl.ndarray.NDArray
-
repeat
public ai.djl.ndarray.NDArray repeat(int axis, long repeats)
- Specified by:
repeat
in interfaceai.djl.ndarray.NDArray
-
repeat
public ai.djl.ndarray.NDArray repeat(long[] repeats)
- Specified by:
repeat
in interfaceai.djl.ndarray.NDArray
-
repeat
public ai.djl.ndarray.NDArray repeat(ai.djl.ndarray.types.Shape desiredShape)
- Specified by:
repeat
in interfaceai.djl.ndarray.NDArray
-
dot
public ai.djl.ndarray.NDArray 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
-
clip
public ai.djl.ndarray.NDArray clip(java.lang.Number min, java.lang.Number max)
- Specified by:
clip
in interfaceai.djl.ndarray.NDArray
-
flip
public ai.djl.ndarray.NDArray flip(int... axes)
- Specified by:
flip
in interfaceai.djl.ndarray.NDArray
-
transpose
public ai.djl.ndarray.NDArray transpose()
- Specified by:
transpose
in interfaceai.djl.ndarray.NDArray
-
transpose
public ai.djl.ndarray.NDArray transpose(int... dimensions)
- Specified by:
transpose
in interfaceai.djl.ndarray.NDArray
-
broadcast
public ai.djl.ndarray.NDArray broadcast(ai.djl.ndarray.types.Shape shape)
- Specified by:
broadcast
in interfaceai.djl.ndarray.NDArray
-
argMax
public ai.djl.ndarray.NDArray argMax()
- Specified by:
argMax
in interfaceai.djl.ndarray.NDArray
-
argMax
public ai.djl.ndarray.NDArray argMax(int axis)
- Specified by:
argMax
in interfaceai.djl.ndarray.NDArray
-
argMin
public ai.djl.ndarray.NDArray argMin()
- Specified by:
argMin
in interfaceai.djl.ndarray.NDArray
-
argMin
public ai.djl.ndarray.NDArray argMin(int axis)
- Specified by:
argMin
in interfaceai.djl.ndarray.NDArray
-
percentile
public ai.djl.ndarray.NDArray percentile(java.lang.Number percentile)
- Specified by:
percentile
in interfaceai.djl.ndarray.NDArray
-
percentile
public ai.djl.ndarray.NDArray percentile(java.lang.Number percentile, int[] dimension)
- Specified by:
percentile
in interfaceai.djl.ndarray.NDArray
-
median
public ai.djl.ndarray.NDArray median()
- Specified by:
median
in interfaceai.djl.ndarray.NDArray
-
median
public ai.djl.ndarray.NDArray median(int[] axes)
- Specified by:
median
in interfaceai.djl.ndarray.NDArray
-
toDense
public ai.djl.ndarray.NDArray toDense()
- Specified by:
toDense
in interfaceai.djl.ndarray.NDArray
-
nonzero
public ai.djl.ndarray.NDArray nonzero()
- Specified by:
nonzero
in interfaceai.djl.ndarray.NDArray
-
getNDArrayInternal
public ai.djl.ndarray.internal.NDArrayEx getNDArrayInternal()
- Specified by:
getNDArrayInternal
in interfaceai.djl.ndarray.NDArray
-
equals
public boolean equals(java.lang.Object obj)
- Overrides:
equals
in classjava.lang.Object
-
hashCode
public int hashCode()
- Overrides:
hashCode
in classjava.lang.Object
-
toString
public java.lang.String toString()
- Overrides:
toString
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<org.tensorflow.internal.c_api.TFE_TensorHandle>
-
setHandle
protected void setHandle(org.tensorflow.internal.c_api.TFE_TensorHandle newHandle)
-
-