Package ai.djl.pytorch.engine

Class PtEngine

  • public final class PtEngine
extends ai.djl.engine.Engine
    The PtEngine is an implementation of the Engine based on the PyTorch Deep Learning Framework.

    To get an instance of the PtEngine when it is not the default Engine, call Engine.getEngine(String) with the Engine name "PyTorch".

    • Field Summary

      Fields 
      Modifier and Type Field Description
      static java.lang.String ENGINE_NAME  

    • Method Summary

      All Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      ai.djl.engine.Engine getAlternativeEngine()
      java.lang.String getEngineName()
      int getRank()
      java.lang.String getVersion()
      boolean hasCapability​(java.lang.String capability)
      ai.djl.ndarray.NDManager newBaseManager()
      ai.djl.ndarray.NDManager newBaseManager​(ai.djl.Device device)
      ai.djl.training.GradientCollector newGradientCollector()
      ai.djl.Model newModel​(java.lang.String name, ai.djl.Device device)
      ai.djl.nn.SymbolBlock newSymbolBlock​(ai.djl.ndarray.NDManager manager)
      void setRandomSeed​(int seed)
      java.lang.String toString()

      • Methods inherited from class ai.djl.engine.Engine

        debugEnvironment, defaultDevice, getAllEngines, getDefaultEngineName, getDevices, getDevices, getEngine, getGpuCount, getInstance, getSeed, hasEngine, newParameterServer, registerEngine

      • Methods inherited from class java.lang.Object

        clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

    • Method Detail

      • getAlternativeEngine

        public ai.djl.engine.Engine getAlternativeEngine()
        Specified by:
        getAlternativeEngine in class ai.djl.engine.Engine

      • getEngineName

        public java.lang.String getEngineName()
        Specified by:
        getEngineName in class ai.djl.engine.Engine

      • getRank

        public int getRank()
        Specified by:
        getRank in class ai.djl.engine.Engine

      • getVersion

        public java.lang.String getVersion()
        Specified by:
        getVersion in class ai.djl.engine.Engine

      • hasCapability

        public boolean hasCapability​(java.lang.String capability)
        Specified by:
        hasCapability in class ai.djl.engine.Engine

      • newSymbolBlock

        public ai.djl.nn.SymbolBlock newSymbolBlock​(ai.djl.ndarray.NDManager manager)
        Specified by:
        newSymbolBlock in class ai.djl.engine.Engine

      • newModel

        public ai.djl.Model newModel​(java.lang.String name,
                             ai.djl.Device device)
        Specified by:
        newModel in class ai.djl.engine.Engine

      • newBaseManager

        public ai.djl.ndarray.NDManager newBaseManager()
        Specified by:
        newBaseManager in class ai.djl.engine.Engine

      • newBaseManager

        public ai.djl.ndarray.NDManager newBaseManager​(ai.djl.Device device)
        Specified by:
        newBaseManager in class ai.djl.engine.Engine

      • newGradientCollector

        public ai.djl.training.GradientCollector newGradientCollector()
        Specified by:
        newGradientCollector in class ai.djl.engine.Engine

      • setRandomSeed

        public void setRandomSeed​(int seed)
        Overrides:
        setRandomSeed in class ai.djl.engine.Engine

      • toString

        public java.lang.String toString()
        Overrides:
        toString in class java.lang.Object