diff --git a/src/brevitas/quant_tensor/__init__.py b/src/brevitas/quant_tensor/__init__.py index e017f5de3..ff1898db2 100644 --- a/src/brevitas/quant_tensor/__init__.py +++ b/src/brevitas/quant_tensor/__init__.py @@ -412,7 +412,7 @@ def __truediv__(self, other): output_signed = self.signed or other.signed output_training = self.training or other.training if self.is_zero_zero_point(self) and self.is_zero_zero_point(other): - output_zero_point = self.zero_point / other.zero_point + output_zero_point = self.zero_point * other.zero_point # Output zero_point is a new, zero-valued tensor else: output_zero_point = None # TODO non-zero zero point output = QuantTensor(