未验证 提交 eb4ad509 编写于 作者: X xiaoting 提交者: GitHub

[doc]Fix maxunpool2d example (#39862)

* fix maxunpool2d example, test=document_fix

* fix maxunpool2d example, test=document_fix
上级 bbe441fc
......@@ -1257,9 +1257,8 @@ class MaxUnPool2D(Layer):
import paddle
import paddle.nn.functional as F
import numpy as np
data = paddle.rand(shape=[1,1,7,7])
data = paddle.rand(shape=[1,1,6,6])
pool_out, indices = F.max_pool2d(data, kernel_size=2, stride=2, padding=0, return_mask=True)
# pool_out shape: [1, 1, 3, 3], indices shape: [1, 1, 3, 3]
Unpool2D = paddle.nn.MaxUnPool2D(kernel_size=2, padding=0)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册