[][src]Crate tch

Modules

data

Dataset iterators.

kind
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.

Kind

Functions

manual_seed
no_grad