Split
概述
算子沿 axis 维度将x拆分成多个张量,张量数量由outputNum指定。
该算子对应的NodeType为NODE_TYPE_SPLIT。
输入:
- x,n维tensor
输出:
- outputs,一组n维张量,每一个张量类型和维度相同,每个张量的类型和x一致。
起始版本: 3.2
相关模块:NNRt
汇总
Public 属性
名称 | 描述 |
---|---|
long outputNum | 指定分割数量。 |
long[] sizeSplits | 指定 输入的张量沿 axis 轴拆分后,每个张量的大小。 如果 sizeSplits 的数据为空,则 sizeSplits 被拆分成大小均等的 张量,此时要求 x.shape[axis] 可以被 outputNum 整除; 如果 sizeSplits 不为空,则要求 sizeSplits 所有元素之和等于 x.shape[axis]。 |
long axis | 指定分割的维度。 |
类成员变量说明
axis
long Split::axis
描述 指定分割的维度。
outputNum
long Split::outputNum
描述 指定分割数量。
sizeSplits
long [] Split::sizeSplits
描述 指定 输入的张量沿 axis 轴拆分后,每个张量的大小。 如果 sizeSplits 的数据为空,则 sizeSplits 被拆分成大小均等的 张量,此时要求 x.shape[axis] 可以被 outputNum 整除; 如果 sizeSplits 不为空,则要求 sizeSplits 所有元素之和等于 x.shape[axis]。