From 1c2273c8e99730b4684c06daebe5528adeeb0155 Mon Sep 17 00:00:00 2001 From: Tongzhou Wang Date: Wed, 5 Dec 2018 11:21:19 -0800 Subject: [PATCH] fix stft arg types Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/14800 Reviewed By: zou3519 Differential Revision: D13340574 Pulled By: SsnL fbshipit-source-id: 8b0dbbe299d1a362da0ecc0b1c0dadb2543ded5d --- torch/functional.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/torch/functional.py b/torch/functional.py index 8757066..b7efdbc 100644 --- a/torch/functional.py +++ b/torch/functional.py @@ -388,10 +388,10 @@ def stft(input, n_fft, hop_length=None, win_length=None, window=None, Arguments: input (Tensor): the input tensor - n_fft (int, optional): size of Fourier transform - hop_length (int): the distance between neighboring sliding window + n_fft (int): size of Fourier transform + hop_length (int, optional): the distance between neighboring sliding window frames. Default: ``None`` (treated as equal to ``floor(n_fft / 4)``) - win_length (int): the size of window frame and STFT filter. + win_length (int, optional): the size of window frame and STFT filter. Default: ``None`` (treated as equal to :attr:`n_fft`) window (Tensor, optional): the optional window function. Default: ``None`` (treated as window of all :math:`1` s) -- 2.7.4