UintxLayout

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

Uintx 張量的佈局類，Uintx 張量是指元素被打包到比標準 8 位位元組更小的位寬的張量。此佈局用於定義資料在 UintxTensor 物件中如何儲存和處理。

變數:

• dtype (torch.dtype) – 張量元素的資料型別，決定了打包使用的位寬。

• pack_dim (int) – 資料打包的維度。預設為 -1，表示最後一個維度。