[ 仅参数名不一致 ]torch.Tensor.unfold

torch.Tensor.unfold

torch.Tensor.unfold(dimension,
                size,
                step)

paddle.Tensor.unfold

paddle.Tensor.unfold(axis,
                size,
                step,
                name=None)

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

参数映射

PyTorch PaddlePaddle 备注
dimension axis 表示需要提取的维度,仅参数名不一致。
size size 表示需要提取的窗口长度。
step step 表示每次提取跳跃的步长。