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