• myq406450149's avatar
    add unbind op (#23359) · eb035f24
    myq406450149 提交于
    * add unbind op
    unbind(tensor, dim=0):
    说明:移除指定维后,返回一组数组,包含了沿着指定维切片后的各个切片。
    
    tensor(Tensor) -- 输入Tensor
    dim(int) -- 删除的维度
    
    示例:
     Input = [[1,2],
               [3,4],
               [5,6]]
      axis = 0
      Output[0] = [1,2]
      Output[1] = [3,4]
      Output[2] = [5,6]
    eb035f24
unbind_op.h 2.5 KB