[][src]Crate tch

Modules

data

Dataset iterators.

index

Indexing operations

kind
nn

A small neural-network library based on Torch.

vision

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

Structs

CModule

A jit PyTorch module.

NewAxis
NoGradGuard

A RAII guard that prevents gradient tracking until deallocated.

Scalar

A single scalar value.

Tensor

A tensor object.

Enums

Cuda

Cuda related helper functions.

Device

A torch device.

IValue

Argument and output values for JIT models.

Kind

The different kind of elements that a Tensor can hold.

Reduction
TensorIndexer

Traits

IndexOp

Functions

manual_seed

Sets the random seed used by torch.

no_grad

Runs a closure without keeping track of gradients.

no_grad_guard

Disables gradient tracking, this will be enabled back when the returned value gets deallocated.