Paddle

Commit Graph

Author	SHA1	Message	Date
yongqiangma	eb035f24d1	add unbind op (#23359 ) * 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]	5 years ago

Author

SHA1

Message

Date

yongqiangma

eb035f24d1

add unbind op (#23359 )

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

1 Commits (8489d4f76f2f05157d1150a7b2e241360d040def)