[ 仅参数名不一致 ] torch.nn.functional.affine_grid

torch.nn.functional.affine_grid

torch.nn.functional.affine_grid(theta,
            size,
            align_corners=None)

paddle.nn.functional.affine_grid

paddle.nn.functional.affine_grid(theta,
            out_shape,
            align_corners=True,
            name=None)

两者功能一致,仅参数名不一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
theta theta 指定仿射变换矩阵
size out_shape 表示指定目标输出图像大小,仅参数名不一致。
align_corners align_corners 指定是否是像素中心对齐