Export TensorIndexer public to candle users (huggingface#2477)

wonkyDD · Sep 13, 2024 · ebf722b · ebf722b
1 parent c09afc2
commit ebf722b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/candle-core/src/lib.rs b/candle-core/src/lib.rs
@@ -81,7 +81,7 @@ pub use custom_op::{CustomOp1, CustomOp2, CustomOp3, InplaceOp1, InplaceOp2, Inp
 pub use device::{Device, DeviceLocation, NdArray};
 pub use dtype::{DType, DTypeParseError, FloatDType, IntDType, WithDType};
 pub use error::{Error, Result};
-pub use indexer::IndexOp;
+pub use indexer::{IndexOp, TensorIndexer};
 pub use layout::Layout;
 pub use shape::{Shape, D};
 pub use storage::Storage;