feat: publish stacked API (#860)

* feat: publish stacked API * feat: publish solely AddDim --------- Co-authored-by: Corey Lowman <[email protected]>
coreylowman · Sep 14, 2023 · 9a208d9 · 9a208d9
1 parent 2e5116c
commit 9a208d9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/tensor_ops/mod.rs b/src/tensor_ops/mod.rs
@@ -264,7 +264,7 @@ pub use slice::slice;
 pub use softmax::softmax;
 pub use sqrt::sqrt;
 pub use square::square;
-pub use stack::TryStack;
+pub use stack::{AddDim, TryStack};
 pub use stddev_to::StddevTo;
 pub use sub::{sub, TrySub};
 pub use sum_to::SumTo;