Enum IndexingPrimsDescriptor 

pub enum IndexingPrimsDescriptor {
    IndexSelect {
        axis: usize,
    },
    Gather {
        axis: usize,
    },
    Scatter {
        axis: usize,
        reduction: ScatterReduction,
    },
    IndexPut {
        accumulate: bool,
    },
}

Descriptor for indexing-family planning.

Examples

use tenferro_prims::{IndexingPrimsDescriptor, ScatterReduction};

let desc = IndexingPrimsDescriptor::IndexSelect { axis: 0 };
assert!(matches!(desc, IndexingPrimsDescriptor::IndexSelect { .. }));

let desc = IndexingPrimsDescriptor::Scatter {
    axis: 1,
    reduction: ScatterReduction::Add,
};
assert!(matches!(desc, IndexingPrimsDescriptor::Scatter { .. }));

Variants

IndexSelect

Select slices along axis using a 1-D index tensor.

Given input of shape [d0, ..., d_{axis}, ..., d_{n-1}] and a 1-D index tensor of length k, produces output of shape [d0, ..., k, ..., d_{n-1}].

Fields

axis: usize

Axis along which to select slices.

Gather

Gather elements along axis using an index tensor of the same rank.

The index tensor shape must match the output shape.

Fields

axis: usize

Axis along which to gather.

Scatter

Scatter source values into output positions along axis.

Inverse of gather: places values from source into output at positions determined by the index tensor.

Fields

axis: usize

Axis along which to scatter.

reduction: ScatterReduction

Reduction mode: overwrite or accumulate.

IndexPut

Put source values at index positions (flat or per-axis).

When accumulate is true, adds into the output instead of overwriting.

Fields

accumulate: bool

Whether to accumulate (add) instead of overwriting.

Trait Implementations

impl Clone for IndexingPrimsDescriptor

fn clone(&self) -> IndexingPrimsDescriptor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for IndexingPrimsDescriptor

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

Formats the value using the given formatter.

impl Hash for IndexingPrimsDescriptor

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for IndexingPrimsDescriptor

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

Tests for self and other values to be equal, and is used by ==.

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

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

impl Eq for IndexingPrimsDescriptor

impl StructuralPartialEq for IndexingPrimsDescriptor