线性 ¶

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

应用线性变换到输入的量化数据： $y = xA^T + b$ 。见 Linear

注意

当前实现每次调用都会打包权重，这会对性能造成惩罚。如果您想避免开销，请使用 Linear 。

参数:

• 输入（张量）- 类型为 torch.quint8 的量化输入

• 权重（张量）- 类型为 torch.qint8 的量化权重

• 偏置（张量）- 无或类型为 torch.float 的 fp32 偏置

• 缩放（双精度浮点数）- 输出缩放。如果为 None，则从输入缩放推导

• 零点（Python：长整型）- 输出零点。如果为 None，则从输入零点推导

返回类型:

张量

形状：

• 输入： $(N, *, in\_features)$ 其中 * 表示任意数量的额外维度

• 重量： $(out\_features, in\_features)$

• 偏见： $(out\_features)$

• 输出： $(N, *, out\_features)$