BlockSparseLayout

class torchao.dtypes.BlockSparseLayout(blocksize: int = 64)

BlockSparseLayout 是一个表示块稀疏矩阵布局的数据类。

变量:

blocksize (int) – 稀疏矩阵中块的大小。默认为 64。