快捷方式

UintxLayout

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

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

變數:
  • dtype (torch.dtype) – 張量元素的資料型別,決定了用於打包的位寬。

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

文件

訪問 PyTorch 的全面開發者文件

檢視文件

教程

獲取針對初學者和高階開發者的深入教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源