unstack

    该OP将单个dim为 D 的Tensor沿 axis 轴unpack为 个dim为 (D-1) 的Tensor

    参数:

    如果 axis < 0 ,则

    返回: 长度为num的Tensor列表, 数据类型与输入Tensor相同,dim为 。

    代码示例