From bdd8107fda4cd02f6a37330ad6c395f70abbdcbc Mon Sep 17 00:00:00 2001 From: Laurent Mazare Date: Sun, 14 Jan 2024 20:09:49 +0100 Subject: [PATCH] Expose the ndarray trait. (#1586) --- candle-core/src/lib.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/candle-core/src/lib.rs b/candle-core/src/lib.rs index 6c4fea91..f2aed1b6 100644 --- a/candle-core/src/lib.rs +++ b/candle-core/src/lib.rs @@ -72,7 +72,7 @@ pub mod utils; mod variable; pub use cpu_backend::CpuStorage; -pub use device::{Device, DeviceLocation}; +pub use device::{Device, DeviceLocation, NdArray}; pub use dtype::{DType, FloatDType, IntDType, WithDType}; pub use error::{Error, Result}; pub use indexer::IndexOp;