[ 仅参数名不一致 ]torch.nn.Flatten¶

torch.nn.Flatten ¶

torch.nn.Flatten(start_dim=1,
                end_dim=-1)

paddle.nn.Flatten(start_axis=1,
                    stop_axis=-1)

两者功能一致且参数用法一致，仅参数名不一致，具体如下：

PyTorch	PaddlePaddle	备注
start_dim	start_axis	展开的起始维度，默认值为 1。
end_dim	stop_axis	展开的结束维度，默认值为 -1。