Package 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 ofNDManager
.
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method 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(ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
ai.djl.ndarray.NDArray
create(java.lang.String[] data, java.nio.charset.Charset charset, ai.djl.ndarray.types.Shape shape)
PtNDArray
create(java.nio.Buffer data, 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)
PtNDArray
from(ai.djl.ndarray.NDArray array)
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
hanningWindow(long numPoints)
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
randomPermutation(long n)
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, attachUncappedInternal, cap, close, copyBuffer, createCSR, createRowSparse, debugDump, defaultDevice, detachInternal, getDevice, getManagedArrays, getName, getParentManager, invoke, invoke, isOpen, load, newSubManager, randomMultinomial, randomMultinomial, sampleGamma, sampleGamma, sampleNormal, sampleNormal, samplePoisson, samplePoisson, setName, tempAttachInternal, toString, truncatedNormal, validateBuffer
-
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, create, create, create, create, createCSR, createCSR, createRowSparse, decode, decode, eye, eye, eye, eye, full, full, full, linspace, linspace, linspace, linspace, load, ones, ones, randomNormal, randomNormal, randomNormal, randomUniform, randomUniform, ret, tempAttachAll, truncatedNormal, truncatedNormal, truncatedNormal, zeros, zeros
-
-
-
-
Method Detail
-
allocateDirect
public java.nio.ByteBuffer allocateDirect(int capacity)
-
from
public PtNDArray from(ai.djl.ndarray.NDArray array)
-
create
public PtNDArray create(ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
- Specified by:
create
in interfaceai.djl.ndarray.NDManager
- Overrides:
create
in classai.djl.ndarray.BaseNDManager
-
create
public PtNDArray create(java.nio.Buffer data, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
-
create
public ai.djl.ndarray.NDArray create(java.lang.String[] data, java.nio.charset.Charset charset, ai.djl.ndarray.types.Shape shape)
- Specified by:
create
in interfaceai.djl.ndarray.NDManager
- Overrides:
create
in classai.djl.ndarray.BaseNDManager
-
createCoo
public ai.djl.ndarray.NDArray createCoo(java.nio.Buffer data, long[][] indices, ai.djl.ndarray.types.Shape shape)
- Specified by:
createCoo
in interfaceai.djl.ndarray.NDManager
- Overrides:
createCoo
in classai.djl.ndarray.BaseNDManager
-
zeros
public ai.djl.ndarray.NDArray zeros(ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
-
ones
public ai.djl.ndarray.NDArray ones(ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
-
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 interfaceai.djl.ndarray.NDManager
- Overrides:
full
in classai.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 interfaceai.djl.ndarray.NDManager
- Overrides:
arange
in classai.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 interfaceai.djl.ndarray.NDManager
- Overrides:
eye
in classai.djl.ndarray.BaseNDManager
-
linspace
public ai.djl.ndarray.NDArray linspace(float start, float stop, int num, boolean endpoint)
- Specified by:
linspace
in interfaceai.djl.ndarray.NDManager
- Overrides:
linspace
in classai.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 interfaceai.djl.ndarray.NDManager
- Overrides:
randomInteger
in classai.djl.ndarray.BaseNDManager
-
randomPermutation
public ai.djl.ndarray.NDArray randomPermutation(long n)
- Specified by:
randomPermutation
in interfaceai.djl.ndarray.NDManager
- Overrides:
randomPermutation
in classai.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 interfaceai.djl.ndarray.NDManager
- Overrides:
randomUniform
in classai.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 interfaceai.djl.ndarray.NDManager
- Overrides:
randomNormal
in classai.djl.ndarray.BaseNDManager
-
hanningWindow
public ai.djl.ndarray.NDArray hanningWindow(long numPoints)
-
newSubManager
public PtNDManager newSubManager(ai.djl.Device device)
-
getEngine
public final ai.djl.engine.Engine getEngine()
-
-