refactor(tensor): crate 项目结构压平

Signed-off-by: YdrMaster <ydrml@hotmail.com>
InfiniTensor · Mar 1, 2024 · 57a9265 · 57a9265
1 parent af02063
commit 57a9265
Showing 7 changed files with 8 additions and 11 deletions.
diff --git a/tensor/src/transform/broadcast.rs → tensor/src/broadcast.rs b/tensor/src/transform/broadcast.rs → tensor/src/broadcast.rs
diff --git a/tensor/src/lib.rs b/tensor/src/lib.rs
@@ -1,22 +1,26 @@
+mod broadcast;
 mod data_type;
 mod fmt;
 mod pattern;
+mod reshape;
+mod slice;
+mod split;
 mod tensor;
-mod transform;
+mod transpose;
 
 #[allow(non_camel_case_types)]
 pub type udim = u32;
 
 #[allow(non_camel_case_types)]
 pub type idim = i32;
 
-use std::mem::{align_of, size_of, size_of_val};
-
 pub use data_type::DataType;
 pub use nalgebra::DVector;
 pub use pattern::{expand_indices, idx_strides, Affine, Shape};
+pub use slice::SliceDim;
 pub use tensor::{Storage, Tensor};
-pub use transform::SliceDim;
+
+use std::mem::{align_of, size_of, size_of_val};
 
 pub fn reslice<T, U>(src: &[T]) -> &[U] {
     let ptr = src.as_ptr_range();

diff --git a/tensor/src/transform/reshape.rs → tensor/src/reshape.rs b/tensor/src/transform/reshape.rs → tensor/src/reshape.rs
diff --git a/tensor/src/transform/slice.rs → tensor/src/slice.rs b/tensor/src/transform/slice.rs → tensor/src/slice.rs
diff --git a/tensor/src/transform/split.rs → tensor/src/split.rs b/tensor/src/transform/split.rs → tensor/src/split.rs
diff --git a/tensor/src/transform/mod.rs b/tensor/src/transform/mod.rs
diff --git a/tensor/src/transform/transpose.rs → tensor/src/transpose.rs b/tensor/src/transform/transpose.rs → tensor/src/transpose.rs