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, 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

    Fields inherited from interface ai.djl.ndarray.NDManager

    UID_GENERATOR

  • Method Summary

    Modifier and Type
    Method
    Description
    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(String[] data, Charset charset, ai.djl.ndarray.types.Shape shape)
    PtNDArray
    create(Buffer data, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
    ai.djl.ndarray.NDArray
    createCoo(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)
    final 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 Details

    • allocateDirect

      public 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(Buffer data, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)

    • create

      public ai.djl.ndarray.NDArray create(String[] data, 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(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)

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

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