Crate tch

source

Modules§

  • Dataset iterators.
  • Indexing operations
  • JIT interface to run model trained/saved using PyTorch Python API.
  • The different kind of elements supported in Torch.
  • A small neural-network library based on Torch.
  • The vision module groups functions and models related to computer vision.

Structs§

Enums§

  • Cuda related helper functions.
  • A torch device.
  • Argument and output values for JIT models. These represent arbitrary values, e.g. tensors, atomic values, pairs of values, etc.
  • The different kind of elements that a Tensor can hold.
  • A tensor layout.
  • Quantization engines
  • Main library error type.

Traits§

Functions§

Type Aliases§