快捷方式

UintxLayout

class torchao.dtypes.UintxLayout(dtype: dtype, pack_dim: int = - 1)[source]

Uintx 张量的布局类,Uintx 张量是指元素被打包到比标准 8 位字节更小的位宽的张量。此布局用于定义数据在 UintxTensor 对象中如何存储和处理。

变量:
  • dtype (torch.dtype) – 张量元素的数据类型,决定了打包使用的位宽。

  • pack_dim (int) – 数据打包的维度。默认为 -1,表示最后一个维度。

文档

访问全面的 PyTorch 开发者文档

查看文档

教程

为初学者和高级开发者提供深入的教程

查看教程

资源

查找开发资源并让您的问题得到解答

查看资源