fix a bug in leakyReLU (#48265)

Summary:
The scale variable needs to be a scalar, otherwise it will report the following error: "RuntimeError: Cannot input a tensor of dimension other than 0 as a scalar argument"

Pull Request resolved: #48265

Test Plan: Tested locally and the error disappeared.

Reviewed By: zhizhengwu

Differential Revision: D25105423

Pulled By: jerryzh168

fbshipit-source-id: 2a0df24cf7e40278a950bffe6e0a9552f99da1d1

torch/nn/quantized/modules/activation.py

@@ -97,8 +97,8 @@ class LeakyReLU(torch.nn.LeakyReLU):
     """
     def __init__(self, scale: float, zero_point: int, negative_slope: float = 1e-2, inplace: bool = False):
         super().__init__(negative_slope, inplace)
-        self.register_buffer('scale', torch.tensor([scale]))
-        self.register_buffer('zero_point', torch.tensor([zero_point]))
+        self.register_buffer('scale', torch.tensor(scale))
+        self.register_buffer('zero_point', torch.tensor(zero_point))
 
     def forward(self, input):
         return torch.ops.quantized.leaky_relu(