[ 仅参数名不一致 ]torch.Tensor.flatten¶
torch.Tensor.flatten¶
torch.Tensor.flatten(start_dim=0, end_dim=- 1)
paddle.Tensor.flatten¶
paddle.Tensor.flatten(start_axis=0, stop_axis=-1, name=None)
两者功能一致且参数用法一致,仅参数名不一致,具体如下:
参数映射¶
PyTorch | PaddlePaddle | 备注 |
---|---|---|
start_dim | start_axis | 展开的起始维度,仅参数名不一致。 |
end_dim | stop_axis | 展开的结束维度,仅参数名不一致。 |