diff --git a/candle-core/src/lib.rs b/candle-core/src/lib.rs index 2a36cebd34..d8d6253213 100644 --- 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;