ai.djl.pytorch.engine

Class PtEngine

java.lang.Object

ai.djl.engine.Engine

ai.djl.pytorch.engine.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 and Description
static java.lang.String	ENGINE_NAME

Method Summary

Modifier and Type	Method and Description
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, getEngine, getInstance, getSeed, hasEngine, newParameterServer

Methods inherited from class java.lang.Object

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

Field Detail

ENGINE_NAME

public static final java.lang.String ENGINE_NAME

See Also:

Constant Field Values

Method Detail

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