harmony 鸿蒙Gather
Gather
概述
根据指定的索引和轴返回输入tensor的切片。
该算子对应的NodeType为NODE_TYPE_GATHER。
输入：
x，n维tensor。
inputIndices，指定输入x在axis上的索引，是一个int类型的数组，值必须在[0,x.shape[axis])范围内。
axis，指定x被切片的轴，int32_t类型的数组，数组长度为1。
输出：
- output，输出切片后的tensor。
Since:
3.2
Version:
1.0
相关模块:
你可能感兴趣的鸿蒙文章
harmony 鸿蒙ActivityRecognitionTypes.idl
0
赞
- 所属分类： 后端技术
- 本文标签：
热门推荐
-
2、 - 优质文章
-
3、 gate.io
-
8、 golang
-
9、 openharmony
-
10、 Vue中input框自动聚焦