Trait tract_hir::prelude::tract_ndarray::NdProducer

pub trait NdProducer {
    type Item;
    type Dim: Dimension;
    fn raw_dim(&self) -> Self::Dim;
    fn __private__(&self) -> PrivateMarker;
}

A producer of an n-dimensional set of elements; for example an array view, mutable array view or an iterator that yields chunks.

Producers are used as a arguments to Zip and azip!().

Comparison to IntoIterator

Most NdProducers are iterable (implement IntoIterator) but not directly iterators. This separation is needed because the producer represents a multidimensional set of items, it can be split along a particular axis for parallelization, and it has no fixed correspondance to a sequence.

The natural exception is one dimensional producers, like AxisIter, which implement Iterator directly (AxisIter traverses a one dimensional sequence, along an axis, while producing multidimensional items).

See also IntoNdProducer

Associated Types

type Item

The element produced per iteration.

type Dim: Dimension

Dimension type

Required methods

fn raw_dim(&self) -> Self::Dim

Return the shape of the producer.

fn __private__(&self) -> PrivateMarker

This trait is private to implement; this method exists to make it impossible to implement outside the crate.

Implementors

impl<'a, A, D> NdProducer for AxisIter<'a, A, D> where
    D: Dimension, 

type Item = <AxisIter<'a, A, D> as Iterator>::Item

type Dim = Dim<[usize; 1]>

type Ptr = *mut A

type Stride = isize

impl<'a, A, D> NdProducer for AxisIterMut<'a, A, D> where
    D: Dimension, 

type Item = <AxisIterMut<'a, A, D> as Iterator>::Item

type Dim = Dim<[usize; 1]>

type Ptr = *mut A

type Stride = isize

impl<'a, A, D> NdProducer for ExactChunks<'a, A, D> where
    D: Dimension, 

type Item = ArrayBase<ViewRepr<&'a A>, D>

type Dim = D

type Ptr = *mut A

type Stride = isize

impl<'a, A, D> NdProducer for ExactChunksMut<'a, A, D> where
    D: Dimension, 

type Item = ArrayBase<ViewRepr<&'a mut A>, D>

type Dim = D

type Ptr = *mut A

type Stride = isize

impl<'a, A, D> NdProducer for Lanes<'a, A, D> where
    D: Dimension, 

type Item = ArrayBase<ViewRepr<&'a A>, Dim<[usize; 1]>>

type Dim = D

type Ptr = *mut A

type Stride = isize

impl<'a, A, D> NdProducer for LanesMut<'a, A, D> where
    D: Dimension, 

type Item = ArrayBase<ViewRepr<&'a mut A>, Dim<[usize; 1]>>

type Dim = D

type Ptr = *mut A

type Stride = isize

impl<'a, A, D> NdProducer for Windows<'a, A, D> where
    D: Dimension, 

type Item = ArrayBase<ViewRepr<&'a A>, D>

type Dim = D

type Ptr = *mut A

type Stride = isize

impl<'a, A, D> NdProducer for ArrayBase<ViewRepr<&'a A>, D> where
    D: Dimension, 

type Item = &'a A

type Dim = D

type Ptr = *mut A

type Stride = isize

impl<'a, A, D> NdProducer for ArrayBase<ViewRepr<&'a mut A>, D> where
    D: Dimension, 

type Item = &'a mut A

type Dim = D

type Ptr = *mut A

type Stride = isize

impl<A, D> NdProducer for ArrayBase<RawViewRepr<*const A>, D> where
    D: Dimension, 

type Item = *const A

type Dim = D

type Ptr = *const A

type Stride = isize

impl<A, D> NdProducer for ArrayBase<RawViewRepr<*mut A>, D> where
    D: Dimension, 

type Item = *mut A

type Dim = D

type Ptr = *mut A

type Stride = isize

impl<D> NdProducer for Indices<D> where
    D: Dimension + Copy, 

type Item = <D as Dimension>::Pattern

type Dim = D

type Ptr = IndexPtr<D>

type Stride = usize