[][src]Crate tch

Re-exports

pub use kind::Kind;

Modules

data

Dataset iterators.

kind

The different kind of elements supported in Torch.

nn

A small neural-network library based on Torch.

vision

The vision module groups functions and models related to computer vision.

Structs

CModule
NoGradGuard
Scalar
Tensor

A tensor object.

Enums

Cpu
Cuda
Device

A torch device.

IValue

Argument and output values for JIT models.

Functions

no_grad