Torch.istft (center=False) Error reported when using window parameter

audio
1 
import torch
import torchaudio
import matplotlib.pyplot as plt


sample_rate = 44100
duration = 5.0
t = torch.arange(0, duration, 1.0 / sample_rate)
waveform = 0.5 * torch.sin(2.0 * 3.14159 * 220.0 * t) + 0.5 * torch.sin(2.0 * 3.14159 * 440.0 * t)

NFFT = 512
hop_size = 256

print("waveform", waveform.shape)

S = torch.stft(waveform, n_fft=NFFT, hop_length=hop_size, win_length=NFFT,
               center=False,
               window=torch.hann_window(window_length=NFFT), return_complex=True)

expected_length = (S.shape[-1] - 1) * hop_size + NFFT+NFFT
centerFalse_wav = torch.istft(S, n_fft=NFFT, hop_length=hop_size, win_length=NFFT,
                              center=False,
                              window=torch.hann_window(window_length=NFFT),
                              length=expected_length)
print(f"reconstructed_waveform shape: {centerFalse_wav.shape}")

line 31, in
centerFalse_wav = torch.istft(S, n_fft=NFFT, hop_length=hop_size, win_length=NFFT,
RuntimeError: istft(CPUComplexFloatType[257, 860], n_fft=512, hop_length=256, win_length=512, window=torch.FloatTensor{[512]}, center=0, normalized=0, onesided=None, length=220928, return_complex=0) window overlap add min: 1
[ CPUBoolType{} ]
If you don’t use window=torch. han_ Window (window_length=NFFT) will not report an error