Struct Strided

pub struct Strided<T> { /* private fields */ }

A single-threaded strided iterator over tensor elements.

Implementations

impl<T: CommonBounds> Strided<T>

pub fn shape(&self) -> &Shape

Retrieves the shape of the tensor.

Returns

A reference to the Shape struct representing the tensor’s dimensions.

pub fn strides(&self) -> &Strides

Retrieves the strides of the tensor.

Returns

A reference to the Strides struct representing the tensor’s stride information.

pub fn new<U: TensorInfo<T>>(tensor: U) -> Self

Creates a new Strided instance from a given tensor.

Arguments

• tensor - The tensor implementing the TensorInfo<T> trait to iterate over mutably.

Returns

A new instance of Strided initialized with the provided tensor.

Trait Implementations

impl<T: Clone> Clone for Strided<T>

fn clone(&self) -> Strided<T>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<T: CommonBounds> IterGetSet for Strided<T>

type Item = T

The type of the iterator’s elements.

fn set_end_index(&mut self, _: usize)

set the end index of the iterator, this is used when rayon perform data splitting

fn set_intervals(&mut self, _: Arc<Vec<(usize, usize)>>)

set the chunk intervals of the iterator, we chunk the outer loop

fn set_strides(&mut self, strides: Strides)

set the strides for the iterator, we call this method normally when we do broadcasting

fn set_shape(&mut self, shape: Shape)

set the shape for the iterator, we call this method normally when we do broadcasting

fn set_prg(&mut self, prg: Vec<i64>)

set the loop progress for the iterator

fn intervals(&self) -> &Arc<Vec<(usize, usize)>>

get the intervals of the iterator

fn strides(&self) -> &Strides

get the strides of the iterator

fn shape(&self) -> &Shape

get the shape of the iterator

fn layout(&self) -> &Layout

get the layout of the iterator

fn broadcast_set_strides(&mut self, shape: &Shape)

set the strides for all the iterators

fn outer_loop_size(&self) -> usize

get the outer loop size

fn inner_loop_size(&self) -> usize

get the inner loop size

fn next(&mut self)

update the loop progress

fn inner_loop_next(&mut self, index: usize) -> Self::Item

get the next element of the inner loop

impl<T: CommonBounds> ShapeManipulator for Strided<T>

fn reshape<S: Into<Shape>>(self, shape: S) -> Self

reshape the iterator, we can change the iteration behavior by changing the shape

fn transpose<AXIS: Into<Axis>>(self, axes: AXIS) -> Self

transpose the iterator, we can change the iteration behavior by changing the axes

fn expand<S: Into<Shape>>(self, shape: S) -> Self

expand the iterator, we can change the iteration behavior by changing the shape

impl<T: CommonBounds> StridedIterator for Strided<T>

fn for_each<F>(self, func: F)

where F: Fn(Self::Item),

perform scalar iteration, this method is for single thread iterator

fn for_each_init<F, INIT, T>(self, init: INIT, func: F)

where F: Fn(&mut T, Self::Item), INIT: Fn() -> T,

perform scalar iteration with init, this method is for single thread iterator

impl<T> StridedIteratorMap for Strided<T>

fn map<'a, T, F, U>(self, f: F) -> StridedMap<'a, Self, T, F>

where F: Fn(T) -> U + Sync + Send + 'a, U: CommonBounds, Self: IterGetSet<Item = T>,

Transforms the strided iterators by applying a provided function to their items.

impl<T> StridedIteratorZip for Strided<T>

fn zip<'a, C>(self, other: C) -> StridedZip<'a, Self, C>

where C: IterGetSet + ShapeManipulator, Self: IterGetSet + ShapeManipulator, <C as IterGetSet>::Item: Send, <Self as IterGetSet>::Item: Send,

Combines this iterator with another iterator, enabling simultaneous iteration.