linear

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

Применяет линейное преобразование к входным квантованным данным: $y = 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, оно вычисляется на основе zero_point входных данных

Тип возвращаемого значения:

Tensor

Форма:

• Входные данные: $(N, *, in\_features)$, где * означает любое количество дополнительных измерений
• Веса: $(out\_features, in\_features)$
• Смещение: $(out\_features)$
• Выходные данные: $(N, *, out\_features)$