diff --git a/crates/kornia-core-ops/src/lib.rs b/crates/kornia-core-ops/src/lib.rs index 4494ce6f..a3a8c878 100644 --- a/crates/kornia-core-ops/src/lib.rs +++ b/crates/kornia-core-ops/src/lib.rs @@ -6,3 +6,6 @@ pub mod error; /// module containing ops implementations. pub mod ops; + +pub use crate::error::TensorOpsError; +pub use crate::ops::TensorOps; diff --git a/crates/kornia-core-ops/src/ops.rs b/crates/kornia-core-ops/src/ops.rs index 05a2d392..01cc4b82 100644 --- a/crates/kornia-core-ops/src/ops.rs +++ b/crates/kornia-core-ops/src/ops.rs @@ -49,7 +49,7 @@ where /// /// ``` /// use kornia_core::{Tensor, CpuAllocator}; - /// use kornia_core_ops::ops::TensorOps; + /// use kornia_core_ops::TensorOps; /// /// let data: [u8; 6] = [1, 1, 1, 1, 1, 1]; /// let t = Tensor::::from_shape_slice([2, 3], &data, CpuAllocator).unwrap(); @@ -80,7 +80,7 @@ where /// /// ``` /// use kornia_core::{Tensor, CpuAllocator}; - /// use kornia_core_ops::ops::TensorOps; + /// use kornia_core_ops::TensorOps; /// /// let data: [f32; 5] = [3., 3., 2., 1., 1.]; /// let t = Tensor::::from_shape_slice([5], &data, CpuAllocator).unwrap(); @@ -110,7 +110,7 @@ where /// /// ``` /// use kornia_core::{Tensor, CpuAllocator}; - /// use kornia_core_ops::ops::TensorOps; + /// use kornia_core_ops::TensorOps; /// /// let data: [f32; 5] = [3., 2., 1., 1., 1.]; /// let t = Tensor::::from_shape_slice([5], &data, CpuAllocator).unwrap();