Add unpool2d op & Expose max_unpool2d API · Pull Request !1589 · PaddlePaddle/Paddle

```
import paddle
import numpy as np
import paddle.nn.functional as F

data = paddle.to_tensor([[[[ 1.,  2,  3,  4],
                            [ 5,  6,  7,  8],
                            [ 9, 10, 11, 12],
                            [13, 14, 15, 16]]]])

output, indices = F.max_pool2d(data, kernel_size=2, stride=2, return_mask=True)
out = F.max_unpool2d(output, indices, kernel_size=2, stride=2,output_size=(5,5))
# paddle: Tensor(shape=[1, 1, 5, 5], dtype=float32, place=CUDAPlace(0), stop_gradient=True,
#        [[[[0. , 0. , 0. , 0. , 0. ],
#           [6. , 0. , 8. , 0. , 0. ],
#           [0. , 0. , 0. , 14., 0. ],
#           [16., 0. , 0. , 0. , 0. ],
#           [0. , 0. , 0. , 0. , 0. ]]]])

```

GVPPaddlePaddle/Paddle

搜索帮助

GVP PaddlePaddle/Paddle