fix doc

PaddlePaddle · Nov 7, 2023 · bf37ce2 · bf37ce2
1 parent dbd9d7f
commit bf37ce2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/paddle/nn/layer/layers.py b/python/paddle/nn/layer/layers.py
@@ -926,7 +926,7 @@ def astype(self, dtype=None):
 
         Parameters:
             dtype(str|paddle.dtype|numpy.dtype): target data type of layer.
-                If set str, it can be "bool",  "float16", "float32", "float64",
+                If set str, it can be "bool", "bfloat16", "float16", "float32", "float64",
                 "int8", "int16", "int32", "int64", "uint8", "complex64", "complex128".
                 Default: None
 
@@ -995,7 +995,7 @@ def astype(self, dtype=None):
             return self
         else:
             raise ValueError(
-                "dtype value error, must be 'bfloat16', 'float16', 'float32', 'float64', 'int8', 'int16', 'int32', 'int64', 'uint8', 'uint16', 'complex64', 'complex128', 'bool', or paddle.dtype, numpy.dtype, but recieve "
+                "dtype value error, must be 'bfloat16', 'float16', 'float32', 'float64', 'int8', 'int16', 'int32', 'int64', 'uint8', 'complex64', 'complex128', 'bool', or paddle.dtype, numpy.dtype, but recieve "
                 + str(dtype)
             )