-
由 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
* 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]