as_view_mut

rust-math · Jun 20, 2020 · 4f803b8 · 4f803b8
1 parent c9e46a9
commit 4f803b8
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> {