def METHOD_NAME(self, x, frame_step, window=None, frame_length=None, onesided=None):  # type: ignore
    if frame_length is None:
        if window is None:
            frame_length = x.shape[-2]
        else:
            frame_length = window.shape[0]
    hop_length = frame_step
    if window is None:
        window = np.ones((frame_length,), dtype=x.dtype)
    n_frames = 1 + (x.shape[-2] - frame_length) // frame_step
    res = _stft(x, [frame_length], hop_length, n_frames, window, onesided=onesided)
    return (res.astype(x.dtype),)