optimize code

Author: yqzhishen

modules/nsf_hifigan/models.py

@@ -249,10 +249,8 @@ def forward(self, f0):
         output uv: tensor(batchsize=1, length, 1)
         """
         with torch.no_grad():
-            f0_buf = torch.zeros(f0.shape[0], f0.shape[1], self.dim,
-                                 device=f0.device)
             # fundamental component
-            fn = torch.multiply(f0, torch.FloatTensor([[range(1, self.harmonic_num + 2)]]).to(f0.device))
+            fn = torch.multiply(f0, torch.arange(1, self.harmonic_num + 2, device=f0.device))
 
             # generate sine waveforms
             sine_waves = self._f02sine(fn) * self.sine_amp