ai.djl.pytorch.engine

Class PtNDManager

java.lang.Object

ai.djl.ndarray.BaseNDManager

ai.djl.pytorch.engine.PtNDManager

All Implemented Interfaces:

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

public class PtNDManager
extends ai.djl.ndarray.BaseNDManager

PtNDManager is the PyTorch implementation of NDManager.

Nested Class Summary

Nested classes/interfaces inherited from class ai.djl.ndarray.BaseNDManager

ai.djl.ndarray.BaseNDManager.TempResource

Field Summary

Fields inherited from class ai.djl.ndarray.BaseNDManager

closed, device, name, parent, resources, tempResources, uid

Method Summary

Modifier and Type	Method and Description
java.nio.ByteBuffer	allocateDirect(int capacity)
ai.djl.ndarray.NDArray	arange(float start, float stop, float step, ai.djl.ndarray.types.DataType dataType)
ai.djl.ndarray.NDArray	arange(int start, int stop, int step, ai.djl.ndarray.types.DataType dataType)
PtNDArray	create(java.nio.Buffer data, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
PtNDArray	create(ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
ai.djl.ndarray.NDArray	createCoo(java.nio.Buffer data, long[][] indices, ai.djl.ndarray.types.Shape shape)
ai.djl.ndarray.NDArray	eye(int rows, int cols, int k, ai.djl.ndarray.types.DataType dataType)
ai.djl.ndarray.NDArray	full(ai.djl.ndarray.types.Shape shape, float value, ai.djl.ndarray.types.DataType dataType)
ai.djl.engine.Engine	getEngine()
ai.djl.ndarray.NDArray	linspace(float start, float stop, int num, boolean endpoint)
PtNDManager	newSubManager(ai.djl.Device device)
ai.djl.ndarray.NDArray	ones(ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
ai.djl.ndarray.NDArray	randomInteger(long low, long high, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
ai.djl.ndarray.NDArray	randomNormal(float loc, float scale, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
ai.djl.ndarray.NDArray	randomUniform(float low, float high, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
ai.djl.ndarray.NDArray	zeros(ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)

Methods inherited from class ai.djl.ndarray.BaseNDManager

attachInternal, close, create, create, createCSR, createRowSparse, debugDump, detachInternal, getDevice, getName, getParentManager, invoke, invoke, isOpen, load, newSubManager, randomMultinomial, randomMultinomial, setName, tempAttachInternal, toString

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Methods inherited from interface ai.djl.ndarray.NDManager

arange, arange, arange, arange, arange, arange, arange, attachAll, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, createCSR, createCSR, createRowSparse, decode, decode, eye, eye, eye, eye, from, full, full, full, linspace, linspace, linspace, linspace, load, newBaseManager, newBaseManager, newBaseManager, ones, ones, randomNormal, randomNormal, randomNormal, randomUniform, randomUniform, ret, tempAttachAll, zeros, zeros

Method Detail

allocateDirect

public java.nio.ByteBuffer allocateDirect(int capacity)

create

public PtNDArray create(ai.djl.ndarray.types.Shape shape,
                        ai.djl.ndarray.types.DataType dataType)

Specified by:

create in interface ai.djl.ndarray.NDManager

Overrides:

create in class ai.djl.ndarray.BaseNDManager

create

public PtNDArray create(java.nio.Buffer data,
                        ai.djl.ndarray.types.Shape shape,
                        ai.djl.ndarray.types.DataType dataType)

createCoo

public ai.djl.ndarray.NDArray createCoo(java.nio.Buffer data,
                                        long[][] indices,
                                        ai.djl.ndarray.types.Shape shape)

Specified by:

createCoo in interface ai.djl.ndarray.NDManager

Overrides:

createCoo in class ai.djl.ndarray.BaseNDManager

zeros

public ai.djl.ndarray.NDArray zeros(ai.djl.ndarray.types.Shape shape,
                                    ai.djl.ndarray.types.DataType dataType)

Specified by:

zeros in interface ai.djl.ndarray.NDManager

Overrides:

zeros in class ai.djl.ndarray.BaseNDManager

ones

public ai.djl.ndarray.NDArray ones(ai.djl.ndarray.types.Shape shape,
                                   ai.djl.ndarray.types.DataType dataType)

Specified by:

ones in interface ai.djl.ndarray.NDManager

Overrides:

ones in class ai.djl.ndarray.BaseNDManager

full

public ai.djl.ndarray.NDArray full(ai.djl.ndarray.types.Shape shape,
                                   float value,
                                   ai.djl.ndarray.types.DataType dataType)

Specified by:

full in interface ai.djl.ndarray.NDManager

Overrides:

full in class ai.djl.ndarray.BaseNDManager

arange

public ai.djl.ndarray.NDArray arange(int start,
                                     int stop,
                                     int step,
                                     ai.djl.ndarray.types.DataType dataType)

arange

public ai.djl.ndarray.NDArray arange(float start,
                                     float stop,
                                     float step,
                                     ai.djl.ndarray.types.DataType dataType)

Specified by:

arange in interface ai.djl.ndarray.NDManager

Overrides:

arange in class ai.djl.ndarray.BaseNDManager

eye

public ai.djl.ndarray.NDArray eye(int rows,
                                  int cols,
                                  int k,
                                  ai.djl.ndarray.types.DataType dataType)

Specified by:

eye in interface ai.djl.ndarray.NDManager

Overrides:

eye in class ai.djl.ndarray.BaseNDManager

linspace

public ai.djl.ndarray.NDArray linspace(float start,
                                       float stop,
                                       int num,
                                       boolean endpoint)

Specified by:

linspace in interface ai.djl.ndarray.NDManager

Overrides:

linspace in class ai.djl.ndarray.BaseNDManager

randomInteger

public ai.djl.ndarray.NDArray randomInteger(long low,
                                            long high,
                                            ai.djl.ndarray.types.Shape shape,
                                            ai.djl.ndarray.types.DataType dataType)

Specified by:

randomInteger in interface ai.djl.ndarray.NDManager

Overrides:

randomInteger in class ai.djl.ndarray.BaseNDManager

randomUniform

public ai.djl.ndarray.NDArray randomUniform(float low,
                                            float high,
                                            ai.djl.ndarray.types.Shape shape,
                                            ai.djl.ndarray.types.DataType dataType)

Specified by:

randomUniform in interface ai.djl.ndarray.NDManager

Overrides:

randomUniform in class ai.djl.ndarray.BaseNDManager

randomNormal

public ai.djl.ndarray.NDArray randomNormal(float loc,
                                           float scale,
                                           ai.djl.ndarray.types.Shape shape,
                                           ai.djl.ndarray.types.DataType dataType)

Specified by:

randomNormal in interface ai.djl.ndarray.NDManager

Overrides:

randomNormal in class ai.djl.ndarray.BaseNDManager

newSubManager

public PtNDManager newSubManager(ai.djl.Device device)

getEngine

public final ai.djl.engine.Engine getEngine()