Trait rten_tensor::MutLayout

source ·

pub trait MutLayout: Layout + Clone {
    // Required methods
    fn from_shape(shape: Self::Index<'_>) -> Self;
    fn from_shape_and_strides(
        shape: Self::Index<'_>,
        strides: Self::Index<'_>,
        overlap: OverlapPolicy
    ) -> Result<Self, FromDataError>;
    fn move_axis(&mut self, from: usize, to: usize);
    fn permuted(&self, order: Self::Index<'_>) -> Self;
    fn resize_dim(&mut self, dim: usize, size: usize);
    fn transposed(&self) -> Self;
    fn slice<const M: usize>(
        &self,
        range: &[SliceItem]
    ) -> (Range<usize>, NdLayout<M>);
    fn slice_dyn(&self, range: &[SliceItem]) -> (Range<usize>, DynLayout);
    fn squeezed(&self) -> DynLayout;
    fn try_slice<R: IntoSliceItems>(
        &self,
        range: R
    ) -> Result<(Range<usize>, DynLayout), SliceError>;

    // Provided method
    fn reshaped<S: IntoLayout>(&self, shape: S) -> S::Layout { ... }
}

Expand description

MutLayout extends Layout with methods for creating, modifying and transforming layouts.

Required Methods§

source

fn from_shape(shape: Self::Index<'_>) -> Self

Create a new contiguous layout with a given shape.

source

fn from_shape_and_strides( shape: Self::Index<'_>, strides: Self::Index<'_>, overlap: OverlapPolicy ) -> Result<Self, FromDataError>

Create a layout with custom strides.

The strides specify the offset gap between successive entries along a given axis. overlap controls whether the layout is allowed to map multiple indices to the same element. This can be true for immutable views, but must be false for tensors or views that are mutable.

source