From 6f18f8cf0cdc44375797bb3557bbc3cd5fe39daf Mon Sep 17 00:00:00 2001
From: KP <109694228@qq.com>
Date: Thu, 3 Mar 2022 14:24:13 +0000
Subject: [PATCH] Add doc.

---
 paddle/fluid/operators/stft_op.cc | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/paddle/fluid/operators/stft_op.cc b/paddle/fluid/operators/stft_op.cc
index f9fd11b9a28394..5a6e6989891263 100644
--- a/paddle/fluid/operators/stft_op.cc
+++ b/paddle/fluid/operators/stft_op.cc
@@ -75,14 +75,18 @@ class StftOp : public framework::OperatorWithKernel {
 class StftOpMaker : public framework::OpProtoAndCheckerMaker {
  public:
   void Make() override {
-    AddInput("X", "");
-    AddOutput("Out", "");
-    AddAttr<int>("n_fft", "");
-    AddAttr<int>("hop_length", "");
-    AddAttr<bool>("normalized", "");
-    AddAttr<bool>("onesided", "");
+    AddInput("X", "Input waveforms with shape (N, T)");
+    AddOutput("Out",
+              "The complex STFT output tensor with shape (N, n_fft, "
+              "num_frames) or (N, n_fft/2 + 1, num_frames)");
+    AddAttr<int>("n_fft", "The number of input samples to perform FFT");
+    AddAttr<int>("hop_length", "Number of samples between adjacent frames");
+    AddAttr<bool>("normalized",
+                  "Control whether to scale the output by 1/sqrt(n_fft)");
+    AddAttr<bool>("onesided",
+                  "Control whether to return half of the FFT output");
     AddComment(R"DOC(
-      Stft Op.
+      Short-time Fourier transform (STFT).
     )DOC");
   }
 };