1. 04 4月, 2020 1 次提交
    • 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