diff --git a/fftw/src/array.rs b/fftw/src/array.rs index 70c2b625..745a1fbe 100644 --- a/fftw/src/array.rs +++ b/fftw/src/array.rs @@ -64,6 +64,14 @@ impl AlignedVec { { ArrayView::from_shape(shape, self.as_slice()) } + + pub fn as_view_mut(&mut self, shape: Shape) -> Result, ShapeError> + where + D: Dimension, + Shape: Into>, + { + ArrayViewMut::from_shape(shape, self.as_slice_mut()) + } } impl Deref for AlignedVec {