Struct Tensor 

pub struct Tensor<T, const N: usize> { /* private fields */ }

张量是一种数据的容器，代表在均质的数据上附加了数据类型、形状和数据布局的动态信息。

作为一个容器，Tensor<T, N> 类似于 Option<T>、Result<T, _>，可通过一系列方法变换其信息或数据：

• clone：复制张量的信息，也复制张量的数据；
• transform：在不改变数据的情况下变换张量的布局；
• map：替换张量的数据；
• as_ref：返回引用原始数据的新张量；
• as_mut：返回可变引用原始数据的新张量；
• …

Implementations

impl<const N: usize> Tensor<usize, N>

pub fn new(dt: DigitLayout, shape: [usize; N]) -> Self

创建使用指定数据类型 dt 和形状 shape 的张量，张量的“数据”是连续存储其数据占用的字节数。

传入的 shape 应为张量中的数值的数量。 在底层存储中，可能将多个数值捆绑为一个数据组。 获取张量的形状时，将返回作为 N 维数组的形状，其连续维度除去了组的规模。

例如，对于将 32 个数字绑定为一组的数据类型，shape 为 [7, 1024] 时，产生的张量的形状是 [7, 32]。

// 定义一个数据类型，以 32 个 8 位无符号数为一组。
digit_layout::layout!(GROUP u(8); 32);

let tensor = Tensor::new(GROUP, [7, 1024]);
assert_eq!(tensor.dt(), GROUP);
assert_eq!(tensor.shape(), [7, 32]);
assert_eq!(tensor.take(), 7 * 32 * 32);

pub fn from_dim_slice(dt: DigitLayout, shape: impl AsRef<[usize]>) -> Self

impl<T, const N: usize> Tensor<T, N>

pub const fn dt(&self) -> DigitLayout

pub const fn layout(&self) -> &ArrayLayout<N>

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

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

pub fn offset(&self) -> isize

pub const fn get(&self) -> &T

pub fn get_mut(&mut self) -> &mut T

pub fn take(self) -> T

pub const fn from_raw_parts( dt: DigitLayout, layout: ArrayLayout<N>, item: T, ) -> Self

pub fn into_raw_parts(self) -> (DigitLayout, ArrayLayout<N>, T)

pub fn use_info(&self) -> Tensor<usize, N>

pub fn is_contiguous(&self) -> bool

impl<T, const N: usize> Tensor<T, N>

pub fn as_ref(&self) -> Tensor<&T, N>

pub fn as_mut(&mut self) -> Tensor<&mut T, N>

pub fn transform(self, f: impl FnOnce(ArrayLayout<N>) -> ArrayLayout<N>) -> Self

pub fn map<U>(self, f: impl FnOnce(T) -> U) -> Tensor<U, N>

pub fn replace<U>(self, u: U) -> (T, Tensor<U, N>)

impl<T: Deref, const N: usize> Tensor<T, N>

pub fn as_deref(&self) -> Tensor<&<T as Deref>::Target, N>

impl<T: DerefMut, const N: usize> Tensor<T, N>

pub fn as_deref_mut(&mut self) -> Tensor<&mut <T as Deref>::Target, N>

Trait Implementations

impl<T: Clone, const N: usize> Clone for Tensor<T, N>

fn clone(&self) -> Tensor<T, N>

Returns a duplicate of the value.

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

Performs copy-assignment from source.