Package org.pytorch

  • Class Summary 
    Class Description
    IValue
    Java representation of a TorchScript value, which is implemented as tagged union that can be one of the supported types: https://pytorch.org/docs/stable/jit.html#types .
    Module
    Java wrapper for torch::jit::script::Module.
    Tensor
    Representation of a Tensor.
  • Enum Summary 
    Enum Description
    DType
    Codes representing tensor data types.