Crate burn_tch 

Burn Tch Backend

Structs

LibTorch

Tensor backend that uses LibTorch with the tch crate for executing tensor operations.

TchShape

A shape that can be used by LibTorch.

TchTensor

A tensor using the tch backend.

Enums

LibTorchDevice

The device struct when using the tch backend.

Storage

A reference to a tensor storage.

Traits

TchElement

The element type for the tch backend.

Functions

dummy_cuda_dependency^⚠

Dummy function to get CUDA to link properly

Type Aliases

StorageRef

A reference to a tensor storage.