Package ai.djl.python.engine
Class PyNDManager
- java.lang.Object
-
- ai.djl.ndarray.BaseNDManager
-
- ai.djl.python.engine.PyNDManager
-
- All Implemented Interfaces:
ai.djl.ndarray.NDManager
,java.lang.AutoCloseable
public class PyNDManager extends ai.djl.ndarray.BaseNDManager
PyNDManager
is the Python engine implementation ofNDManager
.
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description java.nio.ByteBuffer
allocateDirect(int capacity)
void
close()
PyNDArray
create(java.nio.Buffer data, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
PyNDArray
from(ai.djl.ndarray.NDArray array)
ai.djl.engine.Engine
getEngine()
PyNDManager
newSubManager(ai.djl.Device device)
-
Methods inherited from class ai.djl.ndarray.BaseNDManager
arange, attachInternal, copyBuffer, create, create, createCoo, createCSR, createRowSparse, debugDump, defaultDevice, detachInternal, eye, full, getDevice, getName, getParentManager, invoke, invoke, isOpen, linspace, load, newSubManager, ones, randomInteger, randomMultinomial, randomMultinomial, randomNormal, randomUniform, setName, tempAttachInternal, toString, truncatedNormal, validateBufferSize, zeros
-
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, 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 PyNDArray from(ai.djl.ndarray.NDArray array)
-
create
public PyNDArray create(java.nio.Buffer data, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
-
newSubManager
public PyNDManager newSubManager(ai.djl.Device device)
-
getEngine
public final ai.djl.engine.Engine getEngine()
-
close
public void close()
- Specified by:
close
in interfacejava.lang.AutoCloseable
- Specified by:
close
in interfaceai.djl.ndarray.NDManager
- Overrides:
close
in classai.djl.ndarray.BaseNDManager
-
-