harmony 鸿蒙Split
Split
概述
算子沿 axis 维度将x拆分成多个张量，张量数量由outputNum指定。
该算子对应的NodeType为NODE_TYPE_SPLIT。
输入：
- x，n维tensor
输出：
- outputs，一组n维张量，每一个张量类型和维度相同，每个张量的类型和x一致。
Since:
3.2
Version:
1.0
相关模块:
汇总
Public 属性
| 名称 | 描述 |
|---|---|
| outputNum | long 指定分割数量。 |
| sizeSplits | long[] 指定输入的张量沿axis轴拆分后，每个张量的大小。 |
| axis | long 指定分割的维度。 |
类成员变量说明
axis
long Split::axis
描述:
指定分割的维度。
outputNum
long Split::outputNum
描述:
指定分割数量。
sizeSplits
long [] Split::sizeSplits
描述:
指定输入的张量沿 axis 轴拆分后，每个张量的大小。
如果 sizeSplits 的数据为空，则 sizeSplits 被拆分成大小均等的 张量，此时要求 x.shape[axis] 可以被 outputNum 整除；如果 sizeSplits 不为空，则要求 sizeSplits 所有元素之和等于 x.shape[axis]。
你可能感兴趣的鸿蒙文章
harmony 鸿蒙ActivityRecognitionTypes.idl
0
赞
- 所属分类： 后端技术
- 本文标签：
热门推荐
-
2、 - 优质文章
-
3、 gate.io
-
6、 golang
-
7、 openharmony