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

torch.nn.Unflatten ¶

torch.nn.Unflatten(dim, unflattened_size)

paddle.nn.Unflatten(axis, shape, name=None)

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

PyTorch	PaddlePaddle	备注
dim	axis	指定对输入 Tensor 进行运算的轴，仅参数名不一致。
unflattened_size	shape	需要展开的形状，仅参数名不一致。