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::Module.
PyTorchCodegenLoader
Tensor	Representation of a Tensor.

Enum Summary

Enum	Description
Device
DType	Codes representing tensor data types.
MemoryFormat