Package ai.djl.pytorch.jni


package ai.djl.pytorch.jni
Contains classes to interface with the underlying PyTorch Engine.
  • Classes
    Class
    Description
    A 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.