評價此頁

linear#

class torch.ao.nn.quantized.functional.linear(input, weight, bias=None, scale=None, zero_point=None)[source]#

對輸入的量化資料應用線性變換: y=xAT+by = xA^T + b。 請參閱 Linear

注意

當前的實現會在每次呼叫時打包權重,這會影響效能。如果您想避免這種開銷,請使用 Linear

引數
  • input (Tensor) – 型別為 torch.quint8 的量化輸入

  • weight (Tensor) – 型別為 torch.qint8 的量化權重

  • bias (Tensor) – None 或 fp32 偏置,型別為 torch.float

  • scale (double) – 輸出尺度。如果為 None,則從輸入尺度推導

  • zero_point (python:long) – 輸出零點。如果為 None,則從輸入零點推導

返回型別

張量

形狀
  • 輸入: (N,,in_features)(N, *, in\_features),其中 * 表示任意數量的附加維度

  • 權重: (out_features,in_features)(out\_features, in\_features)

  • 偏置: (out_features)(out\_features)

  • 輸出: (N,,out_features)(N, *, out\_features)