burn_tch/
lib.rs

1#![warn(missing_docs)]
2#![cfg_attr(docsrs, feature(doc_cfg))]
3#![allow(clippy::single_range_in_vec_init)]
4
5//! Burn Tch Backend
6
7mod backend;
8mod element;
9mod ops;
10mod tensor;
11
12pub use backend::*;
13pub use element::*;
14pub use tensor::*;
15
16#[cfg(test)]
17mod tests {
18    extern crate alloc;
19
20    type TestBackend = crate::LibTorch<f32>;
21    type TestTensor<const D: usize> = burn_tensor::Tensor<TestBackend, D>;
22    type TestTensorInt<const D: usize> = burn_tensor::Tensor<TestBackend, D, burn_tensor::Int>;
23    type TestTensorBool<const D: usize> = burn_tensor::Tensor<TestBackend, D, burn_tensor::Bool>;
24
25    burn_tensor::testgen_all!();
26    burn_autodiff::testgen_all!();
27}