线性

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

对传入的量化数据应用线性变换：$y = xA^T + b$. 请参阅 Linear

注意

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

参数

• input (张量) – 类型为 torch.quint8 的量化输入

• weight (张量) – 类型为 torch.qint8 的量化权重

• bias (张量) – None 或类型为 torch.float 的 fp32 偏置

• scale (双精度浮点数) – 输出比例。如果为 None，则从输入比例派生

• zero_point (python:long) – 输出零点。如果为 None，则从输入零点派生

返回值类型

张量

形状

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

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

• 偏置：$(out\_features)$

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