harmony 鸿蒙OneHot

  • 2023-06-24
  • 浏览 (454)

OneHot

概述

根据indices指定的位置,生成一个由one-hot向量构成的tensor。

每个onehot向量中的有效值由on_value决定,其他位置由off_value决定。

该算子对应的NodeType为NODE_TYPE_ONE_HOT。

输入:

  • indices,n维tensor。indices中每个元素决定每个one-hot向量,on_value的位置。

  • depth,一个整型标量,决定one-hot向量的深度。要求depth>0。

  • on_value,一个标量,指定one-hot向量中的有效值。

  • off_value,一个标量,指定one-hot向量中除有效位以外,其他位置的值。

输出:

  • output,如果indices时n维tensor,则output是(n+1)维tensor。output的形状由indices和axis共同决定。

Since:

3.2

Version:

1.0

相关模块:

NNRt

汇总

Public 属性

名称 描述
axis long
一个整型标量,指定插入one-hot的维度。

类成员变量说明

axis

long OneHot::axis

描述:

一个整型标量,指定插入one-hot的维度。

indices的形状是[N,C],depth的值是D,当axis=0时,output形状为[D,N,C], indices的形状是[N,C],depth的值是D,当axis=-1时,output形状为[N,C,D], indices的形状是[N,C],depth的值是D,当axis=1时,output形状为[N,D,C]。

你可能感兴趣的鸿蒙文章

harmony 鸿蒙HDI 参数参考

harmony 鸿蒙ActRecognitionEvent

harmony 鸿蒙Activation

harmony 鸿蒙ActivityRecognitionTypes.idl

harmony 鸿蒙AddFusion

harmony 鸿蒙Alignment

harmony 鸿蒙AllocInfo

harmony 鸿蒙ApduSimIORequestInfo

harmony 鸿蒙ArgMaxFusion

harmony 鸿蒙Audio

0  赞