Package ai.djl.pytorch.engine

package ai.djl.pytorch.engine

Contains classes to interface with the underlying PyTorch Engine.

Classes

Class	Description
PtDeviceType	DeviceType is the PyTorch equivalent of the types in Device.
PtEngine	The PtEngine is an implementation of the Engine based on the PyTorch Deep Learning Framework.
PtEngineProvider	PtEngineProvider is the PyTorch implementation of EngineProvider.
PtGradientCollector	PtGradientCollector is the PyTorch implementation of GradientCollector.
PtModel	PtModel is the PyTorch implementation of Model.
PtNDArray	PtNDArray is the PyTorch implementation of NDArray.
PtNDArrayEx	PtNDArrayEx is the PyTorch implementation of the NDArrayEx.
PtNDArrayIndexer	The NDArrayIndexer used by the PtNDArray.
PtNDManager	PtNDManager is the PyTorch implementation of NDManager.
PtSymbolBlock	PtSymbolBlock is the PyTorch implementation of SymbolBlock.