harmony 鸿蒙Gather

2023-06-24 浏览 (958)

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

相关模块:

NNRt

你可能感兴趣的鸿蒙文章

harmony 鸿蒙HDI 参数参考

harmony 鸿蒙ActRecognitionEvent

harmony 鸿蒙Activation

harmony 鸿蒙ActivityRecognitionTypes.idl

harmony 鸿蒙AddFusion

harmony 鸿蒙Alignment

harmony 鸿蒙AllocInfo

harmony 鸿蒙ApduSimIORequestInfo

harmony 鸿蒙ArgMaxFusion

harmony 鸿蒙Audio

  • 所属分类: 后端技术
  • 本文标签: 鸿蒙 软件
  • 版权声明: 本文链接 https://seaxiang.com/blog/a731f5c9597147419efb664e2533a10f