Python PyTorch DeQuantize用法及代碼示例

本文簡要介紹python語言中 torch.nn.quantized.DeQuantize 的用法。

用法:

class torch.nn.quantized.DeQuantize

對傳入的張量進行去量化

例子：：

>>> input = torch.tensor([[1., -1.], [1., -1.]])
>>> scale, zero_point, dtype = 1.0, 2, torch.qint8
>>> qm = Quantize(scale, zero_point, dtype)
>>> quantized_input = qm(input)
>>> dqm = DeQuantize()
>>> dequantized = dqm(quantized_input)
>>> print(dequantized)
tensor([[ 1., -1.],
        [ 1., -1.]], dtype=torch.float32)