未验证 提交 0c4567bb 编写于 作者: 张春乔 提交者: GitHub

[xdoctest] reformat example code with google style in No. 100 (#55967)

* Update backend.py

* transform_parameters

* Update transform_parameters.py

* Update transform_parameters.py

* Update transform_parameters.py

* Update transform_parameters.py

* Update python/paddle/nn/utils/transform_parameters.py

* Apply suggestions from code review

* Apply suggestions from code review
上级 29cc8b1a
...@@ -52,18 +52,18 @@ def _stride_column(param): ...@@ -52,18 +52,18 @@ def _stride_column(param):
Examples: Examples:
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
paddle.seed(100) >>> paddle.seed(100)
linear = paddle.nn.Linear(2, 3) >>> linear = paddle.nn.Linear(2, 3)
print(linear.weight) >>> print(linear.weight)
# [[-0.31485492, -1.02896988, 0.45741916], Parameter containing:
# [-0.65525872, -1.04643178, 1.07262802]] Tensor(shape=[2, 3], dtype=float32, place=Place(cpu), stop_gradient=False,
[[ 0.11732829, -0.64161885, -1.06996548],
[ 0.03456247, -0.29862350, -0.52380574]])
paddle.nn.utils.stride_column(linear.weight) >>> paddle.nn.utils._stride_column(linear.weight)
print(linear.weight) >>> print(linear.weight)
# [[-0.31485492, 0.45741916, -1.04643178],
# [-1.02896988, -0.65525872, 1.07262802]]
""" """
assert len(param.shape) == 2 assert len(param.shape) == 2
...@@ -91,11 +91,13 @@ def parameters_to_vector(parameters, name=None): ...@@ -91,11 +91,13 @@ def parameters_to_vector(parameters, name=None):
Examples: Examples:
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
linear = paddle.nn.Linear(10, 15) >>> paddle.seed(2023)
>>> linear = paddle.nn.Linear(10, 15)
paddle.nn.utils.parameters_to_vector(linear.parameters()) >>> t = paddle.nn.utils.parameters_to_vector(linear.parameters())
# 1-D Tensor: [165] >>> print(t.shape)
[165]
""" """
dtype = parameters[0].dtype dtype = parameters[0].dtype
...@@ -136,19 +138,18 @@ def vector_to_parameters(vec, parameters, name=None): ...@@ -136,19 +138,18 @@ def vector_to_parameters(vec, parameters, name=None):
Examples: Examples:
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
weight_attr = paddle.ParamAttr(initializer=paddle.nn.initializer.Constant(3.)) >>> weight_attr = paddle.ParamAttr(initializer=paddle.nn.initializer.Constant(3.))
linear1 = paddle.nn.Linear(10, 15, weight_attr) >>> linear1 = paddle.nn.Linear(10, 15, weight_attr)
vec = paddle.nn.utils.parameters_to_vector(linear1.parameters()) >>> vec = paddle.nn.utils.parameters_to_vector(linear1.parameters())
linear2 = paddle.nn.Linear(10, 15) >>> linear2 = paddle.nn.Linear(10, 15)
# copy weight of linear1 to linear2 >>> # copy weight of linear1 to linear2
paddle.nn.utils.vector_to_parameters(vec, linear2.parameters()) >>> paddle.nn.utils.vector_to_parameters(vec, linear2.parameters())
# weight: Tensor(shape=[10, 15], dtype=float32, place=CUDAPlace(0), stop_gradient=False, >>> print((linear1.weight == linear2.weight).all())
# [[3. , ..., 3. ], Tensor(shape=[], dtype=bool, place=Place(cpu), stop_gradient=True,
# [..., ..., ...], True)
# [3. , ..., 3. ]])
""" """
origin_shapes = [] origin_shapes = []
sections = [] sections = []
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册