forked from ccf-ai-infra/GPUCodeForces
fixes DequantizeLinear #88
This commit is contained in:
parent
58c49c5bf7
commit
e5ff0163ad
|
|
@ -17,11 +17,6 @@ class Model(nn.Module):
|
|||
self.zero_point = -10.0
|
||||
|
||||
def forward(self, x: torch.Tensor) -> torch.Tensor:
|
||||
"""
|
||||
x: [N] int8
|
||||
Output: [N] float32
|
||||
Formula: (x - zp) * scale
|
||||
"""
|
||||
|
||||
return (x.float() - self.zero_point) * self.scale
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue