harmony 鸿蒙ExpandDims

  • 2025-06-16
  • 浏览 (12)

ExpandDims

概述

在给定轴上为tensor添加一个额外的维度。

该算子对应的NodeType为NODE_TYPE_EXPAND_DIMS。

输入:

  • x,n维tensor。

  • axis,需要添加的维度的index,int32_t类型,值必须在[-dim-1,dim],且只允许常量值。

输出:

  • output,给定轴上添加了额外额度的算子。

起始版本: 3.2

相关模块:NNRt

你可能感兴趣的鸿蒙文章

harmony 鸿蒙NNRt HDI 参考

harmony 鸿蒙Activation

harmony 鸿蒙Activation

harmony 鸿蒙AddFusion

harmony 鸿蒙AddFusion

harmony 鸿蒙ArgMaxFusion

harmony 鸿蒙ArgMaxFusion

harmony 鸿蒙AvgPoolFusion

harmony 鸿蒙AvgPoolFusion

harmony 鸿蒙BatchToSpaceND

0  赞