Package ai.djl.pytorch.engine

Class 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

      • Nested classes/interfaces inherited from interface ai.djl.ndarray.NDManager

        ai.djl.ndarray.NDManager.SystemNDManager

    • Field Summary

      • Fields inherited from class ai.djl.ndarray.BaseNDManager

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

    • 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 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 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)

      • 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 interface ai.djl.ndarray.NDManager
        Overrides:
        create in class ai.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 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

      • randomPermutation

        public ai.djl.ndarray.NDArray randomPermutation​(long n)
        Specified by:
        randomPermutation in interface ai.djl.ndarray.NDManager
        Overrides:
        randomPermutation 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()