未验证 提交 36868e84 编写于 作者: Y yukavio 提交者: GitHub

fix one_hot example doc test=document_fix (#26585)

* fix one_hot example doc test=document_fix

* fix some bug
上级 422a1620
......@@ -72,17 +72,17 @@ def one_hot(x, num_classes, name=None):
Examples:
.. code-block:: python
import paddle.fluid as fluid
import paddle
# Correspond to the first example above, where label.shape is 4 and one_hot_label.shape is [4, 4].
label = fluid.data(name="label", shape=[4, 1], dtype="int64")
label = paddle.data(name="label", shape=[4, 1], dtype="int64")
# label.shape = [4]
# label.data = [1, 1, 3, 0]
one_hot_label = fluid.one_hot(x=label, num_classes=4)
one_hot_label = paddle.nn.functional.one_hot(x=label, num_classes=4)
# one_hot_label.shape = [4, 4]
# one_hot_label.data = [[0., 1., 0., 0.],
[0., 1., 0., 0.],
[0., 0., 0., 1.],
[1., 0., 0., 0.]]
# [0., 1., 0., 0.],
# [0., 0., 0., 1.],
# [1., 0., 0., 0.]]
"""
if in_dygraph_mode():
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册