张量
张量是一种数据的容器,代表在均质的数据上附加了数据类型、形状和数据布局的动态信息。
本项目提供 Tensor<T, N>
结构体,用于灵活管理多维数据,它能通过一系列方法对数据信息和内容进行变换,同时提供了类型安全和高效的操作接口。
使用示例
use Tensor;
use DigitLayout;
use ;
// 定义一个数据类型,以 32 个 8 位无符号数为一组。
layout!;
let shape = ;
let element_size = 32;
let layout = new_contiguous;
let item = 7 * 1024;
// 创建张量。
let tensor = from_raw_parts;
// 获取张量数据类型。
assert_eq!;
// 获取张量布局信息。
assert_eq!;
// 复制张量。
let mut cloned_tensor = tensor.clone;
* += 1;
assert_eq!;