We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 003abc7 commit b94de17Copy full SHA for b94de17
tf2onnx/onnx_opset/signal.py
@@ -22,10 +22,10 @@
22
23
# pylint: disable=unused-argument,missing-docstring
24
25
-def make_dft_constant(N, dtype, fft_length):
26
- n = np.arange(N)
27
- k = n.reshape((N, 1)).astype(np.float64)
28
- mat = np.exp(-2j * np.pi * k * n / N)
+def make_dft_constant(length, dtype, fft_length):
+ n = np.arange(length)
+ k = n.reshape((length, 1)).astype(np.float64)
+ mat = np.exp(-2j * np.pi * k * n / length)
29
mat = mat[:fft_length // 2 + 1]
30
both = np.empty((2,) + mat.shape, dtype=dtype)
31
both[0, :, :] = np.real(mat)
0 commit comments