开发者
资源
[object Object][object Object][object Object]undefined
[object Object]
  • API功能:计算输入张量[object Object]按组归一化的结果,包括张量out、均值meanOut、标准差的倒数rstdOut以及silu的输出。
  • 计算公式:
    • GroupNorm:xx为输入[object Object]γ\gammaβ\beta分别代表输入[object Object][object Object]E[x]=xˉE[x] = \bar{x}代表xx的均值,Var[x]=1ni=1n(xiE[x])2Var[x]=\frac{1}{n}\sum_{i=1}^{n} (x_i - E[x])^2 代表xx的方差,则{groupnormOut=xE[x]Var[x]+epsγ+βmeanOut=E[x]rstdOut=1Var[x]+eps\begin{cases} \text{groupnormOut} = \frac{x - E[x]}{\sqrt{Var[x] + eps}} * \gamma + \beta \\ \text{meanOut} = E[x] \\ \text{rstdOut} = \frac{1}{\sqrt{Var[x] + eps}} \end{cases}
    • Silu:out=groupnormOut1+egroupnormOut\text{out} = \frac{\text{groupnormOut}}{1 + e^{-\text{groupnormOut}}}
[object Object]
[object Object]
[object Object]
  • input ([object Object]):必选参数,源数据张量,维度需要为2~8维且第1维度能整除[object Object]。数据格式支持NDND,支持非连续的Tensor。

    • [object Object]Atlas 推理系列产品[object Object]:数据类型支持[object Object][object Object]
    • [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]:数据类型支持[object Object][object Object][object Object]
  • weight ([object Object]):可选参数,索引张量,维度为1且元素数量需与输入[object Object]的第1维度保持相同,数据格式支持NDND,支持非连续的Tensor。

    • [object Object]Atlas 推理系列产品[object Object]:数据类型支持[object Object][object Object]
    • [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]:数据类型支持[object Object][object Object][object Object]
  • bias ([object Object]):可选参数,更新数据张量,维度为1且元素数量需与输入[object Object]的第1维度保持相同,数据格式支持NDND,支持非连续的Tensor。

    • [object Object]Atlas 推理系列产品[object Object]:数据类型支持[object Object][object Object]
    • [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]:数据类型支持[object Object][object Object][object Object]
  • group ([object Object]):必选参数,表示将输入[object Object]的第1维度分为group组,group需大于0。

  • eps ([object Object]):可选参数,数值稳定性而加到分母上的值,若保持精度,则eps需大于0。默认值为0.00001。

[object Object]
  • out ([object Object]):数据类型和shape与[object Object]相同,支持NDND,支持非连续的Tensor。

    • [object Object]Atlas 推理系列产品[object Object]:数据类型支持[object Object][object Object]
    • [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]:数据类型支持[object Object][object Object][object Object]
  • meanOut ([object Object]):数据类型与[object Object]相同,shape为(N, group),其中N为[object Object]第0维度值。数据格式支持NDND,支持非连续的Tensor。

    • [object Object]Atlas 推理系列产品[object Object]:数据类型支持[object Object][object Object]
    • [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]:数据类型支持[object Object][object Object][object Object]
  • rstdOut ([object Object]):数据类型与[object Object]相同,shape为(N, group),其中N为[object Object]第0维度值。数据格式支持NDND,支持非连续的Tensor。

    • [object Object]Atlas 推理系列产品[object Object]:数据类型支持[object Object][object Object]
    • [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]:数据类型支持[object Object][object Object][object Object]
[object Object]
  • 该接口支持推理、训练场景下使用。
  • [object Object][object Object][object Object][object Object][object Object][object Object]数据类型必须在支持的范围之内。
  • [object Object][object Object][object Object]的数据类型与[object Object]相同;[object Object][object Object][object Object]可以不同。
  • [object Object][object Object]的数据类型必须保持一致,且数据类型的精度不能低于[object Object]的数据类型。
  • [object Object][object Object]的维度需为1且元素数量需与输入[object Object]的第1维度保持相同。
  • [object Object]维度需大于一维且小于等于八维,且[object Object]第1维度能整除[object Object]
  • [object Object]任意维都需大于0。
  • [object Object]的shape与[object Object]相同。
  • [object Object][object Object]的shape为(N, group),其中N为[object Object]第0维度值。
  • [object Object]需大于0。
  • [object Object]需大于0。
[object Object]
[object Object]