as_view_mut

rust-math · Jun 28, 2019 · 2ba2aa0 · 2ba2aa0
1 parent 428e052
commit 2ba2aa0
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/fftw/src/array.rs b/fftw/src/array.rs
@@ -64,6 +64,14 @@ impl<T> AlignedVec<T> {
     {
         ArrayView::from_shape(shape, self.as_slice())
     }
+
+    pub fn as_view_mut<D, Shape>(&mut self, shape: Shape) -> Result<ArrayViewMut<T, D>, ShapeError>
+    where
+        D: Dimension,
+        Shape: Into<StrideShape<D>>,
+    {
+        ArrayViewMut::from_shape(shape, self.as_slice_mut())
+    }
 }
 
 impl<T> Deref for AlignedVec<T> {