torch.nn.quantized ------------------ This module implements the quantized versions of the nn modules and functionals. Functional interface ~~~~~~~~~~~~~~~~~~~~ .. automodule:: torch.nn.quantized.functional .. autofunction:: linear .. autofunction:: conv1d .. autofunction:: conv2d .. autofunction:: conv3d .. autofunction:: max_pool2d .. autofunction:: adaptive_avg_pool2d .. autofunction:: avg_pool2d .. autofunction:: interpolate .. autofunction:: hardswish .. autofunction:: upsample .. autofunction:: upsample_bilinear .. autofunction:: upsample_nearest .. automodule:: torch.nn.quantized ReLU6 ~~~~~~~~~~~~~~~ .. autoclass:: ReLU6 :members: ELU ~~~~~~~~~~~~~~~ .. autoclass:: ELU :members: Hardswish ~~~~~~~~~~~~~~~ .. autoclass:: Hardswish :members: Conv1d ~~~~~~~~~~~~~~~ .. autoclass:: Conv1d :members: Conv2d ~~~~~~~~~~~~~~~ .. autoclass:: Conv2d :members: Conv3d ~~~~~~~~~~~~~~~ .. autoclass:: Conv3d :members: FloatFunctional ~~~~~~~~~~~~~~~ .. autoclass:: FloatFunctional :members: QFunctional ~~~~~~~~~~~~~~~ .. autoclass:: QFunctional :members: Quantize ~~~~~~~~~~~~~~~ .. autoclass:: Quantize :members: DeQuantize ~~~~~~~~~~~~~~~ .. autoclass:: DeQuantize :members: Linear ~~~~~~~~~~~~~~~ .. autoclass:: Linear :members: BatchNorm2d ~~~~~~~~~~~~~~~ .. autoclass:: BatchNorm2d :members: BatchNorm3d ~~~~~~~~~~~~~~~ .. autoclass:: BatchNorm3d :members: LayerNorm ~~~~~~~~~~~~~~~ .. autoclass:: LayerNorm :members: GroupNorm ~~~~~~~~~~~~~~~ .. autoclass:: GroupNorm :members: InstanceNorm1d ~~~~~~~~~~~~~~~ .. autoclass:: InstanceNorm1d :members: InstanceNorm2d ~~~~~~~~~~~~~~~ .. autoclass:: InstanceNorm2d :members: InstanceNorm3d ~~~~~~~~~~~~~~~ .. autoclass:: InstanceNorm3d :members: Embedding ~~~~~~~~~~~~~~~ .. autoclass:: Embedding :members: EmbeddingBag ~~~~~~~~~~~~~~~ .. autoclass:: EmbeddingBag :members: