[ 仅参数名不一致 ]torch.nn.Unflatten¶
torch.nn.Unflatten¶
torch.nn.Unflatten(dim, unflattened_size)
参数映射¶
PyTorch | PaddlePaddle | 备注 |
---|---|---|
dim | axis | 指定对输入 Tensor 进行运算的轴,仅参数名不一致。 |
unflattened_size | shape | 需要展开的形状,仅参数名不一致。 |
torch.nn.Unflatten(dim, unflattened_size)
PyTorch | PaddlePaddle | 备注 |
---|---|---|
dim | axis | 指定对输入 Tensor 进行运算的轴,仅参数名不一致。 |
unflattened_size | shape | 需要展开的形状,仅参数名不一致。 |