Struct ArrayLayout

Source

pub struct ArrayLayout<const N: usize> { /* private fields */ }

Expand description

An array layout allow N dimensions inlined.

Implementations§

Source §

impl<const N: usize> ArrayLayout<N>

Source

pub fn broadcast(&self, axis: usize, times: usize) -> Self

广播变换将指定的长度为 1 的阶扩增指定的倍数，并将其步长固定为 0。

let layout = ArrayLayout::<3>::new(&[1, 5, 2], &[10, 2, 1], 0).broadcast(0, 10);
assert_eq!(layout.shape(), &[10, 5, 2]);
assert_eq!(layout.strides(), &[0, 2, 1]);
assert_eq!(layout.offset(), 0);

Source

pub fn broadcast_many(&self, args: &[BroadcastArg]) -> Self

一次对多个阶进行广播变换。

Source §

impl<const N: usize> ArrayLayout<N>

Source

pub fn index(&self, axis: usize, index: usize) -> Self

索引变换是选择张量指定阶上一项数据的变换，例如指定向量中的一个数、指定矩阵的一行或一列。索引变换导致张量降阶，确定索引的阶从张量表示移除。

let layout = ArrayLayout::<3>::new(&[2, 3, 4], &[12, 4, 1], 0).index(1, 2);
assert_eq!(layout.shape(), &[2, 4]);
assert_eq!(layout.strides(), &[12, 1]);
assert_eq!(layout.offset(), 8);

Source

pub fn index_many(&self, args: &[IndexArg]) -> Self

一次对多个阶进行索引变换。

Source §

impl<const N: usize> ArrayLayout<N>

Source

pub fn merge(&self, range: Range<usize>) -> Option<Self>

合并变换是将多个连续维度划分合并的变换。

let layout = ArrayLayout::<3>::new(&[2, 3, 4], &[12, 4, 1], 0).merge(0..3).unwrap();
assert_eq!(layout.shape(), &[24]);
assert_eq!(layout.strides(), &[1]);
assert_eq!(layout.offset(), 0);

Source

pub fn merge_many(&self, args: &[Range<usize>]) -> Option<Self>

一次对多个阶进行合并变换。

Source §

impl<const N: usize> ArrayLayout<N>

Source

pub fn slice(&self, axis: usize, start: usize, step: isize, len: usize) -> Self

切片变换是裁剪张量指定阶上一组连续数据的变换。

// axis = 1, start = 1, step = -1, len = 2
let layout = ArrayLayout::<3>::new(&[2, 3, 4], &[12, 4, 1], 0).slice(1, 2, -1, 2);
assert_eq!(layout.shape(), &[2, 2, 4]);
assert_eq!(layout.strides(), &[12, -4, 1]);
assert_eq!(layout.offset(), 8);

Source

pub fn slice_many(&self, args: &[SliceArg]) -> Self

一次对多个阶进行切片变换。

Source §

impl<const N: usize> ArrayLayout<N>

Source

pub fn split<'a>(&'a self, axis: usize, parts: &'a [usize]) -> Split<'a, N> ⓘ

切分变换讲单个张量沿某个维度切分成多个张量，因此可以支持不均匀的切分。

let layout = ArrayLayout::<3>::new(&[2, 3, 4], &[12, 4, 1], 0);
let mut splits = layout.split(2, &[1, 3]);

let layout = splits.next().unwrap();
assert_eq!(layout.shape(), &[2, 3, 1]);
assert_eq!(layout.strides(), &[12, 4, 1]);
assert_eq!(layout.offset(), 0);

let layout = splits.next().unwrap();
assert_eq!(layout.shape(), &[2, 3, 3]);
assert_eq!(layout.strides(), &[12, 4, 1]);
assert_eq!(layout.offset(), 1);

Source §

impl<const N: usize> ArrayLayout<N>

Source

pub fn tile_be(&self, axis: usize, tiles: &[usize]) -> Self

分块变换是将单个维度划分为多个分块的变换。大端分块使得分块后范围更大的维度在形状中更靠前的位置。

let layout = ArrayLayout::<3>::new(&[2, 3, 6], &[18, 6, 1], 0).tile_be(2, &[2, 3]);
assert_eq!(layout.shape(), &[2, 3, 2, 3]);
assert_eq!(layout.strides(), &[18, 6, 3, 1]);
assert_eq!(layout.offset(), 0);

Source

pub fn tile_le(&self, axis: usize, tiles: &[usize]) -> Self

分块变换是将单个维度划分为多个分块的变换。小端分块使得分块后范围更小的维度在形状中更靠前的位置。

let layout = ArrayLayout::<3>::new(&[2, 3, 6], &[18, 6, 1], 0).tile_le(2, &[2, 3]);
assert_eq!(layout.shape(), &[2, 3, 2, 3]);
assert_eq!(layout.strides(), &[18, 6, 1, 2]);
assert_eq!(layout.offset(), 0);

Source

pub fn tile_many(&self, args: &[TileArg<'_>]) -> Self

一次对多个阶进行分块变换。

Source §

impl<const N: usize> ArrayLayout<N>

Source

pub fn transpose(&self, perm: &[usize]) -> Self

转置变换允许调换张量的维度顺序，但不改变元素的存储顺序。

let layout = ArrayLayout::<3>::new(&[2, 3, 4], &[12, 4, 1], 0).transpose(&[1, 0]);
assert_eq!(layout.shape(), &[3, 2, 4]);
assert_eq!(layout.strides(), &[4, 12, 1]);
assert_eq!(layout.offset(), 0);

Source §

impl<const N: usize> ArrayLayout<N>

Source

pub fn new(shape: &[usize], strides: &[isize], offset: isize) -> Self

Creates a new Layout with the given shape, strides, and offset.

let layout = ArrayLayout::<4>::new(&[2, 3, 4], &[12, -4, 1], 20);
assert_eq!(layout.offset(), 20);
assert_eq!(layout.shape(), &[2, 3, 4]);
assert_eq!(layout.strides(), &[12, -4, 1]);

Source

pub fn new_contiguous( shape: &[usize], endian: Endian, element_size: usize, ) -> Self

Creates a new contiguous Layout with the given shape.

let layout = ArrayLayout::<4>::new_contiguous(&[2, 3, 4], Endian::LittleEndian, 4);
assert_eq!(layout.offset(), 0);
assert_eq!(layout.shape(), &[2, 3, 4]);
assert_eq!(layout.strides(), &[4, 8, 24]);