SemiSparseLayout

类 torchao.dtypes.SemiSparseLayout

SemiSparseLayout 是一个布局类，用于处理仿射量化张量中的半结构化稀疏矩阵。此布局专门设计用于 2:4 稀疏模式，即每四个元素中有两个被剪枝为零。此类提供了预处理输入张量以符合此稀疏模式的方法。