修复 paddle.argmax 等 API 的文档

python/paddle/device/__init__.py

@@ -252,6 +252,9 @@ def set_device(device):
             It can be ``cpu``, ``gpu``, ``xpu``, ``npu``, ``gpu:x``, ``xpu:x``, ``npu:x`` and ``ipu``,
             where ``x`` is the index of the GPUs, XPUs or NPUs.
 
+    Returns:
+        Place,the Place to set.
+
     Examples:
 
      .. code-block:: python

python/paddle/hapi/callbacks.py

@@ -882,7 +882,7 @@ def on_eval_end(self, logs=None):
 
 class VisualDL(Callback):
     """
-    VisualDL callback function.
+    VisualDL callback class. After storing the loss values and evaluation metrics in a log file during the training time , the panel is launched to view the visual results.
 
     Args:
         log_dir (str): The directory to save visualdl log file.

python/paddle/nn/layer/container.py

@@ -521,6 +521,9 @@ def extend(self, sublayers):
         Parameters:
             sublayers (iterable of Layer): iterable of sublayers to append
 
+        Returns:
+            None
+
         Examples:
             .. code-block:: python
 

python/paddle/nn/layer/transformer.py

@@ -1461,7 +1461,7 @@ def generate_square_subsequent_mask(self, length):
             length (int|Tensor): The length of sequence.
 
         Returns:
-            Tensor: Generated square mask according to the given length.
+            Tensor, generated square mask according to the given length. The shape is [length, length].
 
         Examples:
             .. code-block:: python

python/paddle/tensor/creation.py

@@ -156,10 +156,11 @@ def create_parameter(
 
     Args:
         shape (list of int): Shape of the parameter
-        dtype (str): Data type of the parameter
+        dtype (str): Data type of the parameter. It can be set as 'float16', 'float32', 'float64'.
         name (str, optional): For detailed information, please refer to
            :ref:`api_guide_Name` . Usually name is no need to set and None by default.
-        attr (ParamAttr, optional): Attributes of the parameter
+        attr (ParamAttr, optional): Attribute object of the specified argument. For detailed information, please refer to
+           :ref:`api_paddle_ParamAttr` None by default, which means that ParamAttr will be initialized as it is.
         is_bias (bool, optional): This can affect which default initializer is chosen
                        when default_initializer is None. If is_bias,
                        initializer.Constant(0.0) will be used. Otherwise,