fix

mthrok · mthrok · commit de4d2e9fc150 · 2020-04-20T18:51:54.000Z
diff --git a/aten/src/ATen/native/SpectralOps.cpp b/aten/src/ATen/native/SpectralOps.cpp
@@ -348,7 +348,7 @@ Tensor istft(const Tensor& self, const int64_t n_fft, const optional<int64_t> ho
 
   Tensor input = self;
   if (input_dim == 3) {
-    input = input.unsqueeze();
+    input = input.unsqueeze(0);
   }
 
   input = input.transpose(1, 2);  // size: (channel, n_frames, fft_size, 2)
diff --git a/test/test_torch.py b/test/test_torch.py
@@ -12263,7 +12263,7 @@ def _test_istft_is_inverse_of_stft(stft_kwargs):
                     # trim the original for case when constructed signal is shorter than original
                     original = original[..., :inversed.size(-1)]
                     self.assertEqual(
-                        inversed, original, 7e-6, 'istft comparison against original', exact_dtype=True)
+                        inversed, original, 'istft comparison against original', atol=7e-6, exact_dtype=True)
 
         patterns = [
             # hann_window, centered, normalized, onesided

Original file line number	Diff line number	Diff line change
`@@ -348,7 +348,7 @@ Tensor istft(const Tensor& self, const int64_t n_fft, const optional<int64_t> ho`
`348`	`348`
`349`	`349`	`Tensor input = self;`
`350`	`350`	`if (input_dim == 3) {`
`351`		`- input = input.unsqueeze();`
	`351`	`+ input = input.unsqueeze(0);`
`352`	`352`	`}`
`353`	`353`
`354`	`354`	`input = input.transpose(1, 2); // size: (channel, n_frames, fft_size, 2)`