[ 仅 API 调用方式不一致 ]torch.nn.Unfold

torch.nn.Unfold

torch.nn.Unfold(kernel_size, dilation=1, padding=0, stride=1)

paddle.compat.nn.Unfold

paddle.compat.nn.Unfold(kernel_size, dilation=1, padding=0, stride=1)

两者功能一致,但调用方式不一致,具体如下:

转写示例

# PyTorch 写法
unfold = torch.nn.Unfold(kernel_size=(2, 2))

# Paddle 写法
unfold = paddle.compat.nn.Unfold(kernel_size=(2, 2))