Package ai.djl.tensorflow.engine

Class TfNDManager

  • All Implemented Interfaces:
    ai.djl.ndarray.NDManager, java.lang.AutoCloseable
    public class TfNDManager
extends ai.djl.ndarray.BaseNDManager

    • 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 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)
      TfNDArray create​(java.nio.Buffer data, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
      ai.djl.ndarray.NDArray createStringTensor​(ai.djl.ndarray.types.Shape shape, java.nio.ByteBuffer... data)
      Creates a String NDArray based on the provided shape.
      ai.djl.ndarray.NDArray eye​(int rows, int cols, int k, ai.djl.ndarray.types.DataType dataType)
      TfNDArray 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)
      org.tensorflow.internal.c_api.TFE_Context getEagerSession()  
      ai.djl.engine.Engine getEngine()
      ai.djl.ndarray.NDArray linspace​(float start, float stop, int num, boolean endpoint)
      TfNDManager newSubManager​(ai.djl.Device device)
      ai.djl.ndarray.NDArray ones​(ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
      ai.djl.tensorflow.engine.TfOpExecutor opExecutor​(java.lang.String operation)  
      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 truncatedNormal​(float loc, float scale, 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, createCoo, createCSR, createRowSparse, debugDump, defaultDevice, detachInternal, getDevice, getManagedArrays, getName, getParentManager, invoke, invoke, isOpen, load, newSubManager, randomInteger, randomMultinomial, randomMultinomial, randomPermutation, sampleGamma, sampleGamma, sampleNormal, sampleNormal, samplePoisson, samplePoisson, setName, tempAttachInternal, toString, 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, 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 TfNDArray from​(ai.djl.ndarray.NDArray array)

      • create

        public ai.djl.ndarray.NDArray 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 TfNDArray 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

      • createStringTensor

        public ai.djl.ndarray.NDArray createStringTensor​(ai.djl.ndarray.types.Shape shape,
                                                 java.nio.ByteBuffer... data)
        Creates a String NDArray based on the provided shape.
        Parameters:
        shape - the shape of the String NDArray
        data - the flattened String array
        Returns:
        a new instance of NDArray

      • getEngine

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

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

      • 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

      • truncatedNormal

        public ai.djl.ndarray.NDArray truncatedNormal​(float loc,
                                              float scale,
                                              ai.djl.ndarray.types.Shape shape,
                                              ai.djl.ndarray.types.DataType dataType)
        Specified by:
        truncatedNormal in interface ai.djl.ndarray.NDManager
        Overrides:
        truncatedNormal in class ai.djl.ndarray.BaseNDManager

      • newSubManager

        public TfNDManager newSubManager​(ai.djl.Device device)

      • getEagerSession

        public org.tensorflow.internal.c_api.TFE_Context getEagerSession()

      • opExecutor

        public ai.djl.tensorflow.engine.TfOpExecutor opExecutor​(java.lang.String operation)