今天看啥  ›  专栏  ›  zartbot

Tensor-009 Cute Tensor

zartbot  · 公众号  ·  · 2024-09-09 19:09

文章预览

前一篇我们详细介绍了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分 ………………………………

原文地址:访问原文地址
快照地址: 访问文章快照
总结与预览地址:访问总结与预览