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框自动聚焦