I try to train this function, as weight and bias a autogrand = True, Then I got
      7     def forward(self, x):
----> 8         x = torch.round(x / self.scale) * self.scale
      9         x = x * self.conv_weight + self.conv_bias
     10         return x

RuntimeError: round_vml_cpu not implemented for 'Long'