torchtraining.quantization

Quantization related operations.

class torchtraining.quantization.Dequantize

Given a quantized Tensor, dequantize it and return float Tensor.

Parameters

data (torch.Tensor) – Quantized torch.Tensor to dequantize.

forward(data)

Returns

data – Dequantized to float32 torch.Tensor

Return type

torch.Tensor