[ 仅参数名不一致 ]torch.unflatten
torch.unflatten
torch.unflatten(input, dim, sizes)
参数映射
| PyTorch | PaddlePaddle | 备注 |
|---|---|---|
| input | x | 输入 Tensor,仅参数名不一致。 |
| dim | axis | 需要变换的维度,仅参数名不一致。 |
| sizes | shape | 维度变换的新形状,仅参数名不一致。 |

torch.unflatten(input, dim, sizes)
| PyTorch | PaddlePaddle | 备注 |
|---|---|---|
| input | x | 输入 Tensor,仅参数名不一致。 |
| dim | axis | 需要变换的维度,仅参数名不一致。 |
| sizes | shape | 维度变换的新形状,仅参数名不一致。 |