roll

    该OP沿着指定维度 axis 对输入 进行循环滚动，当元素移动到最后位置时，会从第一个位置重新插入。如果 axis 为 ，则输入在被循环滚动之前，会先展平成 1-D Tensor ，滚动操作完成后恢复成原来的形状。

    参数：

    • axis (int|list|tuple, optinal) – 滚动轴。

    返回：

    • Tensor，数据类型同输入。

    代码示例：