huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 11:37:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #127 from LaurentMazare/tensor_indexing

Browse Source 
`i(..)` indexing sugar (partial).
This commit is contained in:
Nicolas Patry
2023-07-10 19:56:34 +02:00
committed by GitHub
parent f29b77ec19 1ad235953b
commit fba07d6b6b
3 changed files with 246 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-core/src/lib.rs
View File

@ -43,6 +43,7 @@ pub mod display;
mod dtype;
mod dummy_cuda_backend;
mod error;
mod indexer;
mod layout;
#[cfg(feature = "mkl")]
mod mkl;
@ -59,6 +60,7 @@ pub use cpu_backend::CpuStorage;
pub use device::{Device, DeviceLocation};
pub use dtype::{DType, WithDType};
pub use error::{Error, Result};
pub use indexer::IndexOp;
pub use layout::Layout;
pub use shape::{Shape, D};
pub use storage::Storage;