文章预览
前一篇我们详细介绍了CuTe Layout及其相关的代数, 这一篇我们将开始介绍Tensor. 简单来看Layout只是定义了元素编排和底层存储之间的位置关系, 但是并没有关联到真正的存储. 给一个相对狭义的Tensor定义, 它是通过物理存储空间和Layout定义的一种数据结构, 对外暴露出一种多维数组的形式,对内基于Layout进行索引. 对于一个Tensor由Layout和Engine两个模版参数表示, Layout已经在前一篇详细介绍了, 它是一个逻辑结构将坐标映射到偏移量(Offset), 而Engine是一个基于Offset和解引用的迭代器. 本文目录如下: 1. 创建Tensor 1.1 张量所有权 1.1 .1 Non-Owning Tensor 1.1 .2 Owning Tensor 1.2 小结 2. 使用张量 2.1 基本操作 2.2 元素访问 2.3 张量切片(Slicing) 2.4 Flatten/Coalesce/Group_modes 3. Tiling & Partitioning 3.1 Tensor除法 3.2 Partitioning 3.2 .1 Inner分
………………………………