Package ai.djl.pytorch.jni
Contains classes to interface with the underlying PyTorch Engine.
-
Class Summary Class Description IValue A class represent a PyTorchIValue
data.IValueUtils IValueUtils is utility class to deal with IValue in PyTorch.JniUtils A class containing utilities to interact with the PyTorch Engine's Java Native Interface (JNI) layer.LibUtils Utilities for finding the PyTorch Engine binary on the System.