[ 仅参数名不一致 ]torch.flatten

torch.flatten

torch.flatten(input,
              start_dim=0,
              end_dim=-1)

paddle.flatten

paddle.flatten(x,
               start_axis=0,
               stop_axis=-1,
               name=None)

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

参数映射

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | input | x | 表示输入的 Tensor ,仅参数名不一致。 | | start_dim | start_axis | 表示 flatten 展开的起始维度,仅参数名不一致。 | | end_dim | stop_axis | 表示 flatten 展开的结束维度,仅参数名不一致。 |