Skip to main content

StridedLayout

Struct StridedLayout 

Source
pub struct StridedLayout { /* private fields */ }
Expand description

Strided tensor layout for efficient views and slicing

Implementations§

Source§

impl StridedLayout

Source

pub fn new(shape: Vec<usize>) -> Self

Create a new strided layout with default C-contiguous strides

Source

pub fn with_strides( shape: Vec<usize>, strides: Vec<isize>, offset: usize, ) -> Result<Self>

Create layout with custom strides

Source

pub fn shape(&self) -> &[usize]

Get the shape

Source

pub fn strides(&self) -> &[isize]

Get the strides

Source

pub fn offset(&self) -> usize

Get the offset

Source

pub fn numel(&self) -> usize

Get total number of elements

Source

pub fn is_contiguous(&self) -> bool

Check if the layout is contiguous (C-order)

Source

pub fn is_fortran_contiguous(&self) -> bool

Check if the layout is Fortran contiguous (F-order)

Source

pub fn linear_index(&self, indices: &[usize]) -> Result<usize>

Compute the linear index for a multi-dimensional index

Source

pub fn slice(&self, ranges: &[Range<usize>]) -> Result<Self>

Create a view by slicing along dimensions

Source

pub fn slice_with_stride(&self, slice_params: &[SliceParams]) -> Result<Self>

Create a view by slicing along dimensions with stride support

Source

pub fn transpose(&self, axes: Option<&[usize]>) -> Result<Self>

Transpose the layout

Source

pub fn reshape(&self, new_shape: Vec<usize>) -> Result<Self>

Reshape the layout (only works if contiguous)

Source

pub fn broadcast_to(&self, target_shape: &[usize]) -> Result<Self>

Broadcast to a new shape

Source

pub fn indices_iter(&self) -> StridedIndicesIter

Create an iterator over all valid indices

Trait Implementations§

Source§

impl Clone for StridedLayout

Source§

fn clone(&self) -> StridedLayout

Returns a duplicate of the value. Read more
1.0.0 · Source§

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source. Read more
Source§

impl Debug for StridedLayout

Source§

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter. Read more
Source§

impl PartialEq for StridedLayout

Source§

fn eq(&self, other: &StridedLayout) -> bool

Tests for self and other values to be equal, and is used by ==.
1.0.0 · Source§

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.
Source§

impl Eq for StridedLayout

Source§

impl StructuralPartialEq for StridedLayout

Auto Trait Implementations§

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<T> CloneToUninit for T
where T: Clone,

Source§

unsafe fn clone_to_uninit(&self, dest: *mut u8)

🔬This is a nightly-only experimental API. (clone_to_uninit)
Performs copy-assignment from self to dest. Read more
Source§

impl<Q, K> Equivalent<K> for Q
where Q: Eq + ?Sized, K: Borrow<Q> + ?Sized,

Source§

fn equivalent(&self, key: &K) -> bool

Compare self to key and return true if they are equal.
Source§

impl<Q, K> Equivalent<K> for Q
where Q: Eq + ?Sized, K: Borrow<Q> + ?Sized,

Source§

fn equivalent(&self, key: &K) -> bool

Checks if this value is equivalent to the given key. Read more
Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> IntoEither for T

Source§

fn into_either(self, into_left: bool) -> Either<Self, Self>

Converts self into a Left variant of Either<Self, Self> if into_left is true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more
Source§

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
where F: FnOnce(&Self) -> bool,

Converts self into a Left variant of Either<Self, Self> if into_left(&self) returns true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more
Source§

impl<T> Pointable for T

Source§

const ALIGN: usize

The alignment of pointer.
Source§

type Init = T

The type for initializers.
Source§

unsafe fn init(init: <T as Pointable>::Init) -> usize

Initializes a with the given initializer. Read more
Source§

unsafe fn deref<'a>(ptr: usize) -> &'a T

Dereferences the given pointer. Read more
Source§

unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

Mutably dereferences the given pointer. Read more
Source§

unsafe fn drop(ptr: usize)

Drops the object pointed to by the given pointer. Read more
Source§

impl<T> ToOwned for T
where T: Clone,

Source§

type Owned = T

The resulting type after obtaining ownership.
Source§

fn to_owned(&self) -> T

Creates owned data from borrowed data, usually by cloning. Read more
Source§

fn clone_into(&self, target: &mut T)

Uses borrowed data to replace owned data, usually by cloning. Read more
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
Source§

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

Source§

fn vzip(self) -> V