ai.djl.pytorch.engine

Class PtNDArray

java.lang.Object

ai.djl.pytorch.jni.NativeResource

ai.djl.pytorch.engine.PtNDArray

All Implemented Interfaces:

ai.djl.ndarray.NDArray, java.lang.AutoCloseable

public class PtNDArray
extends NativeResource
implements ai.djl.ndarray.NDArray

PtNDArray is the PyTorch implementation of NDArray.

Field Summary

Fields inherited from class ai.djl.pytorch.jni.NativeResource

handle

Method Summary

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	attachGradient()
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	createMask(ai.djl.ndarray.index.NDIndex index)
PtNDArray	createMask(java.util.function.Predicate<java.lang.Number> predicate)
PtNDArray	cumSum()
PtNDArray	cumSum(int axis)
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 other)
boolean	equals(java.lang.Object obj)
PtNDArray	exp()
PtNDArray	expandDims(int axis)
PtNDArray	flatten()
PtNDArray	floor()
PtNDArray	get(ai.djl.ndarray.index.NDIndex index)
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 other)
PtNDArray	gte(ai.djl.ndarray.NDArray other)
PtNDArray	gte(java.lang.Number other)
int	hashCode()
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[] axes, float temperature)
PtNDArray	lt(ai.djl.ndarray.NDArray other)
PtNDArray	lt(java.lang.Number other)
PtNDArray	lte(ai.djl.ndarray.NDArray other)
PtNDArray	lte(java.lang.Number 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 other)
PtNDArray	none()
PtNDArray	nonzero()
PtNDArray	onesLike()
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)
ai.djl.ndarray.NDArray	reshapeLike(ai.djl.ndarray.NDArray array)
PtNDArray	round()
void	set(java.nio.Buffer data)
void	set(ai.djl.ndarray.index.NDIndex index, ai.djl.ndarray.NDArray value)
void	set(ai.djl.ndarray.index.NDIndex index, java.lang.Number value)
void	setName(java.lang.String name)
void	setScalar(ai.djl.ndarray.index.NDIndex index, java.lang.Number value)
PtNDArray	sin()
PtNDArray	sinh()
PtNDArray	softmax(int[] axes, float temperature)
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)
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()
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()
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()
PtNDArray	zerosLike()

Methods inherited from class ai.djl.pytorch.jni.NativeResource

finalize, getHandle, getUid, isReleased

Methods inherited from class java.lang.Object

clone, getClass, notify, notifyAll, wait, wait, wait

Methods inherited from interface ai.djl.ndarray.NDArray

allClose, allClose, argSort, argSort, attach, booleanMask, broadcast, concat, concat, countNonzero, countNonzero, detach, get, get, get, getBoolean, getByte, getDouble, getFloat, getInt, getLong, getScalar, getUid, getUint8, isEmpty, isScalar, isSparse, like, logSoftmax, logSoftmax, logSoftmax, max, mean, min, prod, reshape, set, set, set, set, set, shapeEquals, size, size, softmax, softmax, softmax, split, split, stack, stack, sum, toArray, toBooleanArray, toByteArray, toDebugString, toDoubleArray, toFloatArray, toIntArray, toLongArray, toUint8Array, trace, trace

Method Detail

getManager

public PtNDManager getManager()

Specified by:

getManager in interface ai.djl.ndarray.NDArray

getName

public java.lang.String getName()

Specified by:

getName in interface ai.djl.ndarray.NDArray

setName

public void setName(java.lang.String name)

Specified by:

setName in interface ai.djl.ndarray.NDArray

getDataType

public ai.djl.ndarray.types.DataType getDataType()

Specified by:

getDataType in interface ai.djl.ndarray.NDArray

getDevice

public ai.djl.Device getDevice()

Specified by:

getDevice in interface ai.djl.ndarray.NDArray

getShape

public ai.djl.ndarray.types.Shape getShape()

Specified by:

getShape in interface ai.djl.ndarray.NDArray

getSparseFormat

public ai.djl.ndarray.types.SparseFormat getSparseFormat()

Specified by:

getSparseFormat in interface ai.djl.ndarray.NDArray

toDevice

public PtNDArray toDevice(ai.djl.Device device,
                          boolean copy)

Specified by:

toDevice in interface ai.djl.ndarray.NDArray

toType

public PtNDArray toType(ai.djl.ndarray.types.DataType dataType,
                        boolean copy)

Specified by:

toType in interface ai.djl.ndarray.NDArray

attachGradient

public void attachGradient()

Specified by:

attachGradient in interface ai.djl.ndarray.NDArray

getGradient

public PtNDArray getGradient()

Specified by:

getGradient in interface ai.djl.ndarray.NDArray

toByteBuffer

public java.nio.ByteBuffer toByteBuffer()

Specified by:

toByteBuffer in interface ai.djl.ndarray.NDArray

set

public void set(java.nio.Buffer data)

Specified by:

set in interface ai.djl.ndarray.NDArray

set

public void set(ai.djl.ndarray.index.NDIndex index,
                ai.djl.ndarray.NDArray value)

Specified by:

set in interface ai.djl.ndarray.NDArray

set

public void set(ai.djl.ndarray.index.NDIndex index,
                java.lang.Number value)

Specified by:

set in interface ai.djl.ndarray.NDArray

setScalar

public void setScalar(ai.djl.ndarray.index.NDIndex index,
                      java.lang.Number value)

Specified by:

setScalar in interface ai.djl.ndarray.NDArray

get

public PtNDArray get(ai.djl.ndarray.index.NDIndex index)

Specified by:

get in interface ai.djl.ndarray.NDArray

copyTo

public void copyTo(ai.djl.ndarray.NDArray array)

Specified by:

copyTo in interface ai.djl.ndarray.NDArray

duplicate

public ai.djl.ndarray.NDArray duplicate()

Specified by:

duplicate in interface ai.djl.ndarray.NDArray

booleanMask

public PtNDArray booleanMask(ai.djl.ndarray.NDArray index,
                             int axis)

Specified by:

booleanMask in interface ai.djl.ndarray.NDArray

zerosLike

public PtNDArray zerosLike()

Specified by:

zerosLike in interface ai.djl.ndarray.NDArray

onesLike

public PtNDArray onesLike()

Specified by:

onesLike in interface ai.djl.ndarray.NDArray

contentEquals

public boolean contentEquals(java.lang.Number number)

Specified by:

contentEquals in interface ai.djl.ndarray.NDArray

contentEquals

public boolean contentEquals(ai.djl.ndarray.NDArray other)

Specified by:

contentEquals in interface ai.djl.ndarray.NDArray

eq

public PtNDArray eq(java.lang.Number other)

Specified by:

eq in interface ai.djl.ndarray.NDArray

eq

public PtNDArray eq(ai.djl.ndarray.NDArray other)

Specified by:

eq in interface ai.djl.ndarray.NDArray

neq

public PtNDArray neq(java.lang.Number other)

Specified by:

neq in interface ai.djl.ndarray.NDArray

neq

public PtNDArray neq(ai.djl.ndarray.NDArray other)

Specified by:

neq in interface ai.djl.ndarray.NDArray

gt

public PtNDArray gt(java.lang.Number other)

Specified by:

gt in interface ai.djl.ndarray.NDArray

gt

public PtNDArray gt(ai.djl.ndarray.NDArray other)

Specified by:

gt in interface ai.djl.ndarray.NDArray

gte

public PtNDArray gte(java.lang.Number other)

Specified by:

gte in interface ai.djl.ndarray.NDArray

gte

public PtNDArray gte(ai.djl.ndarray.NDArray other)

Specified by:

gte in interface ai.djl.ndarray.NDArray

lt

public PtNDArray lt(java.lang.Number other)

Specified by:

lt in interface ai.djl.ndarray.NDArray

lt

public PtNDArray lt(ai.djl.ndarray.NDArray other)

Specified by:

lt in interface ai.djl.ndarray.NDArray

lte

public PtNDArray lte(java.lang.Number other)

Specified by:

lte in interface ai.djl.ndarray.NDArray

lte

public PtNDArray lte(ai.djl.ndarray.NDArray other)

Specified by:

lte in interface ai.djl.ndarray.NDArray

add

public PtNDArray add(java.lang.Number n)

Specified by:

add in interface ai.djl.ndarray.NDArray

add

public PtNDArray add(ai.djl.ndarray.NDArray other)

Specified by:

add in interface ai.djl.ndarray.NDArray

sub

public PtNDArray sub(java.lang.Number n)

Specified by:

sub in interface ai.djl.ndarray.NDArray

sub

public PtNDArray sub(ai.djl.ndarray.NDArray other)

Specified by:

sub in interface ai.djl.ndarray.NDArray

mul

public PtNDArray mul(java.lang.Number n)

Specified by:

mul in interface ai.djl.ndarray.NDArray

mul

public PtNDArray mul(ai.djl.ndarray.NDArray other)

Specified by:

mul in interface ai.djl.ndarray.NDArray

div

public PtNDArray div(java.lang.Number n)

Specified by:

div in interface ai.djl.ndarray.NDArray

div

public PtNDArray div(ai.djl.ndarray.NDArray other)

Specified by:

div in interface ai.djl.ndarray.NDArray

mod

public PtNDArray mod(java.lang.Number n)

Specified by:

mod in interface ai.djl.ndarray.NDArray

mod

public PtNDArray mod(ai.djl.ndarray.NDArray other)

Specified by:

mod in interface ai.djl.ndarray.NDArray

pow

public PtNDArray pow(java.lang.Number n)

Specified by:

pow in interface ai.djl.ndarray.NDArray

pow

public PtNDArray pow(ai.djl.ndarray.NDArray other)

Specified by:

pow in interface ai.djl.ndarray.NDArray

addi

public PtNDArray addi(java.lang.Number n)

Specified by:

addi in interface ai.djl.ndarray.NDArray

addi

public PtNDArray addi(ai.djl.ndarray.NDArray other)

Specified by:

addi in interface ai.djl.ndarray.NDArray

subi

public PtNDArray subi(java.lang.Number n)

Specified by:

subi in interface ai.djl.ndarray.NDArray

subi

public PtNDArray subi(ai.djl.ndarray.NDArray other)

Specified by:

subi in interface ai.djl.ndarray.NDArray

muli

public PtNDArray muli(java.lang.Number n)

Specified by:

muli in interface ai.djl.ndarray.NDArray

muli

public PtNDArray muli(ai.djl.ndarray.NDArray other)

Specified by:

muli in interface ai.djl.ndarray.NDArray

divi

public PtNDArray divi(java.lang.Number n)

Specified by:

divi in interface ai.djl.ndarray.NDArray

divi

public PtNDArray divi(ai.djl.ndarray.NDArray other)

Specified by:

divi in interface ai.djl.ndarray.NDArray

modi

public PtNDArray modi(java.lang.Number n)

Specified by:

modi in interface ai.djl.ndarray.NDArray

modi

public PtNDArray modi(ai.djl.ndarray.NDArray other)

Specified by:

modi in interface ai.djl.ndarray.NDArray

powi

public PtNDArray powi(java.lang.Number n)

Specified by:

powi in interface ai.djl.ndarray.NDArray

powi

public PtNDArray powi(ai.djl.ndarray.NDArray other)

Specified by:

powi in interface ai.djl.ndarray.NDArray

maximum

public PtNDArray maximum(java.lang.Number n)

Specified by:

maximum in interface ai.djl.ndarray.NDArray

maximum

public PtNDArray maximum(ai.djl.ndarray.NDArray other)

Specified by:

maximum in interface ai.djl.ndarray.NDArray

minimum

public PtNDArray minimum(java.lang.Number n)

Specified by:

minimum in interface ai.djl.ndarray.NDArray

minimum

public PtNDArray minimum(ai.djl.ndarray.NDArray other)

Specified by:

minimum in interface ai.djl.ndarray.NDArray

all

public PtNDArray all()

Specified by:

all in interface ai.djl.ndarray.NDArray

any

public PtNDArray any()

Specified by:

any in interface ai.djl.ndarray.NDArray

none

public PtNDArray none()

Specified by:

none in interface ai.djl.ndarray.NDArray

neg

public PtNDArray neg()

Specified by:

neg in interface ai.djl.ndarray.NDArray

negi

public PtNDArray negi()

Specified by:

negi in interface ai.djl.ndarray.NDArray

abs

public PtNDArray abs()

Specified by:

abs in interface ai.djl.ndarray.NDArray

square

public PtNDArray square()

Specified by:

square in interface ai.djl.ndarray.NDArray

sqrt

public ai.djl.ndarray.NDArray sqrt()

Specified by:

sqrt in interface ai.djl.ndarray.NDArray

cbrt

public PtNDArray cbrt()

Specified by:

cbrt in interface ai.djl.ndarray.NDArray

floor

public PtNDArray floor()

Specified by:

floor in interface ai.djl.ndarray.NDArray

ceil

public PtNDArray ceil()

Specified by:

ceil in interface ai.djl.ndarray.NDArray

round

public PtNDArray round()

Specified by:

round in interface ai.djl.ndarray.NDArray

trunc

public PtNDArray trunc()

Specified by:

trunc in interface ai.djl.ndarray.NDArray

exp

public PtNDArray exp()

Specified by:

exp in interface ai.djl.ndarray.NDArray

log

public PtNDArray log()

Specified by:

log in interface ai.djl.ndarray.NDArray

log10

public PtNDArray log10()

Specified by:

log10 in interface ai.djl.ndarray.NDArray

log2

public PtNDArray log2()

Specified by:

log2 in interface ai.djl.ndarray.NDArray

sin

public PtNDArray sin()

Specified by:

sin in interface ai.djl.ndarray.NDArray

cos

public PtNDArray cos()

Specified by:

cos in interface ai.djl.ndarray.NDArray

tan

public PtNDArray tan()

Specified by:

tan in interface ai.djl.ndarray.NDArray

asin

public PtNDArray asin()

Specified by:

asin in interface ai.djl.ndarray.NDArray

acos

public PtNDArray acos()

Specified by:

acos in interface ai.djl.ndarray.NDArray

atan

public PtNDArray atan()

Specified by:

atan in interface ai.djl.ndarray.NDArray

sinh

public PtNDArray sinh()

Specified by:

sinh in interface ai.djl.ndarray.NDArray

cosh

public PtNDArray cosh()

Specified by:

cosh in interface ai.djl.ndarray.NDArray

tanh

public PtNDArray tanh()

Specified by:

tanh in interface ai.djl.ndarray.NDArray

asinh

public PtNDArray asinh()

Specified by:

asinh in interface ai.djl.ndarray.NDArray

acosh

public PtNDArray acosh()

Specified by:

acosh in interface ai.djl.ndarray.NDArray

atanh

public PtNDArray atanh()

Specified by:

atanh in interface ai.djl.ndarray.NDArray

toDegrees

public PtNDArray toDegrees()

Specified by:

toDegrees in interface ai.djl.ndarray.NDArray

toRadians

public PtNDArray toRadians()

Specified by:

toRadians in interface ai.djl.ndarray.NDArray

max

public PtNDArray max()

Specified by:

max in interface ai.djl.ndarray.NDArray

max

public PtNDArray max(int[] axes,
                     boolean keepDims)

Specified by:

max in interface ai.djl.ndarray.NDArray

min

public PtNDArray min()

Specified by:

min in interface ai.djl.ndarray.NDArray

min

public PtNDArray min(int[] axes,
                     boolean keepDims)

Specified by:

min in interface ai.djl.ndarray.NDArray

sum

public PtNDArray sum()

Specified by:

sum in interface ai.djl.ndarray.NDArray

sum

public PtNDArray sum(int[] axes,
                     boolean keepDims)

Specified by:

sum in interface ai.djl.ndarray.NDArray

prod

public PtNDArray prod()

Specified by:

prod in interface ai.djl.ndarray.NDArray

prod

public PtNDArray prod(int[] axes,
                      boolean keepDims)

Specified by:

prod in interface ai.djl.ndarray.NDArray

mean

public PtNDArray mean()

Specified by:

mean in interface ai.djl.ndarray.NDArray

mean

public PtNDArray mean(int[] axes,
                      boolean keepDims)

Specified by:

mean in interface ai.djl.ndarray.NDArray

trace

public PtNDArray trace(int offset,
                       int axis1,
                       int axis2)

Specified by:

trace in interface ai.djl.ndarray.NDArray

split

public ai.djl.ndarray.NDList split(long sections,
                                   int axis)

Specified by:

split in interface ai.djl.ndarray.NDArray

split

public ai.djl.ndarray.NDList split(long[] indices,
                                   int axis)

Specified by:

split in interface ai.djl.ndarray.NDArray

flatten

public PtNDArray flatten()

Specified by:

flatten in interface ai.djl.ndarray.NDArray

reshape

public PtNDArray reshape(ai.djl.ndarray.types.Shape shape)

Specified by:

reshape in interface ai.djl.ndarray.NDArray

reshapeLike

public ai.djl.ndarray.NDArray reshapeLike(ai.djl.ndarray.NDArray array)

Specified by:

reshapeLike in interface ai.djl.ndarray.NDArray

expandDims

public PtNDArray expandDims(int axis)

Specified by:

expandDims in interface ai.djl.ndarray.NDArray

squeeze

public PtNDArray squeeze()

Specified by:

squeeze in interface ai.djl.ndarray.NDArray

squeeze

public PtNDArray squeeze(int axis)

Specified by:

squeeze in interface ai.djl.ndarray.NDArray

squeeze

public PtNDArray squeeze(int[] axes)

Specified by:

squeeze in interface ai.djl.ndarray.NDArray

logicalAnd

public PtNDArray logicalAnd(ai.djl.ndarray.NDArray other)

Specified by:

logicalAnd in interface ai.djl.ndarray.NDArray

logicalOr

public PtNDArray logicalOr(ai.djl.ndarray.NDArray other)

Specified by:

logicalOr in interface ai.djl.ndarray.NDArray

logicalXor

public PtNDArray logicalXor(ai.djl.ndarray.NDArray other)

Specified by:

logicalXor in interface ai.djl.ndarray.NDArray

logicalNot

public PtNDArray logicalNot()

Specified by:

logicalNot in interface ai.djl.ndarray.NDArray

argSort

public PtNDArray argSort(int axis,
                         boolean ascending)

Specified by:

argSort in interface ai.djl.ndarray.NDArray

sort

public PtNDArray sort()

Specified by:

sort in interface ai.djl.ndarray.NDArray

sort

public PtNDArray sort(int axis)

Specified by:

sort in interface ai.djl.ndarray.NDArray

softmax

public PtNDArray softmax(int[] axes,
                         float temperature)

Specified by:

softmax in interface ai.djl.ndarray.NDArray

logSoftmax

public PtNDArray logSoftmax(int[] axes,
                            float temperature)

Specified by:

logSoftmax in interface ai.djl.ndarray.NDArray

cumSum

public PtNDArray cumSum()

Specified by:

cumSum in interface ai.djl.ndarray.NDArray

cumSum

public PtNDArray cumSum(int axis)

Specified by:

cumSum in interface ai.djl.ndarray.NDArray

isInfinite

public PtNDArray isInfinite()

Specified by:

isInfinite in interface ai.djl.ndarray.NDArray

isNaN

public PtNDArray isNaN()

Specified by:

isNaN in interface ai.djl.ndarray.NDArray

createMask

public PtNDArray createMask(ai.djl.ndarray.index.NDIndex index)

Specified by:

createMask in interface ai.djl.ndarray.NDArray

createMask

public PtNDArray createMask(java.util.function.Predicate<java.lang.Number> predicate)

Specified by:

createMask in interface ai.djl.ndarray.NDArray

tile

public PtNDArray tile(long repeats)

Specified by:

tile in interface ai.djl.ndarray.NDArray

tile

public PtNDArray tile(int axis,
                      long repeats)

Specified by:

tile in interface ai.djl.ndarray.NDArray

tile

public PtNDArray tile(long[] repeats)

Specified by:

tile in interface ai.djl.ndarray.NDArray

tile

public PtNDArray tile(ai.djl.ndarray.types.Shape desiredShape)

Specified by:

tile in interface ai.djl.ndarray.NDArray

repeat

public PtNDArray repeat(long repeats)

Specified by:

repeat in interface ai.djl.ndarray.NDArray

repeat

public PtNDArray repeat(int axis,
                        long repeats)

Specified by:

repeat in interface ai.djl.ndarray.NDArray

repeat

public PtNDArray repeat(long[] repeats)

Specified by:

repeat in interface ai.djl.ndarray.NDArray

repeat

public PtNDArray repeat(ai.djl.ndarray.types.Shape desiredShape)

Specified by:

repeat in interface ai.djl.ndarray.NDArray

dot

public PtNDArray dot(ai.djl.ndarray.NDArray other)

Specified by:

dot in interface ai.djl.ndarray.NDArray

clip

public PtNDArray clip(java.lang.Number min,
                      java.lang.Number max)

Specified by:

clip in interface ai.djl.ndarray.NDArray

swapAxes

public PtNDArray swapAxes(int axis1,
                          int axis2)

Specified by:

swapAxes in interface ai.djl.ndarray.NDArray

transpose

public PtNDArray transpose()

Specified by:

transpose in interface ai.djl.ndarray.NDArray

transpose

public PtNDArray transpose(int... axes)

Specified by:

transpose in interface ai.djl.ndarray.NDArray

broadcast

public PtNDArray broadcast(ai.djl.ndarray.types.Shape shape)

Specified by:

broadcast in interface ai.djl.ndarray.NDArray

argMax

public PtNDArray argMax()

Specified by:

argMax in interface ai.djl.ndarray.NDArray

argMax

public PtNDArray argMax(int axis)

Specified by:

argMax in interface ai.djl.ndarray.NDArray

argMin

public PtNDArray argMin()

Specified by:

argMin in interface ai.djl.ndarray.NDArray

argMin

public PtNDArray argMin(int axis)

Specified by:

argMin in interface ai.djl.ndarray.NDArray

percentile

public PtNDArray percentile(java.lang.Number percentile)

Specified by:

percentile in interface ai.djl.ndarray.NDArray

percentile

public PtNDArray percentile(java.lang.Number percentile,
                            int[] axes)

Specified by:

percentile in interface ai.djl.ndarray.NDArray

median

public PtNDArray median()

Specified by:

median in interface ai.djl.ndarray.NDArray

median

public PtNDArray median(int[] axes)

Specified by:

median in interface ai.djl.ndarray.NDArray

toDense

public PtNDArray toDense()

Specified by:

toDense in interface ai.djl.ndarray.NDArray

toSparse

public PtNDArray toSparse(ai.djl.ndarray.types.SparseFormat fmt)

Specified by:

toSparse in interface ai.djl.ndarray.NDArray

nonzero

public PtNDArray nonzero()

Specified by:

nonzero in interface ai.djl.ndarray.NDArray

getNDArrayInternal

public PtNDArrayEx getNDArrayInternal()

Specified by:

getNDArrayInternal in interface ai.djl.ndarray.NDArray

toString

public java.lang.String toString()

Overrides:

toString in class java.lang.Object

equals

public boolean equals(java.lang.Object obj)

Overrides:

equals in class java.lang.Object

hashCode

public int hashCode()

Overrides:

hashCode in class java.lang.Object

close

public void close()

Specified by:

close in interface ai.djl.ndarray.NDArray

Specified by:

close in interface java.lang.AutoCloseable

Overrides:

close in class NativeResource