文件最后提交记录最后更新时间
5 个月前
5 个月前
2 个月前
5 个月前
5 个月前
5 个月前
2 个月前
README

InstanceNorm

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT
Atlas A3 训练系列产品/Atlas A3 推理系列产品 ×
Atlas A2 训练系列产品/Atlas A2 推理系列产品 ×
Atlas 200I/500 A2 推理产品 ×
Atlas 推理系列产品 ×
Atlas 训练系列产品 ×

功能说明

  • 算子功能:用于执行Instance Normalization(实例归一化)操作。与BatchNorm相比,InstanceNorm在每个样本的实例上进行归一化,而不是在整个批次上进行归一化,这使得该函数更适合处理图像等数据。

  • 计算公式:

    y=x−E(x)Var(x)+ε∗γ+βy = {{x-E(x)}\over\sqrt {Var(x) + ε}} * γ + β

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入
  • 支持空Tensor,仅支持reduce轴输入为0,不支持N轴和C轴输入为0。
  • 表示进行InstanceNorm计算的输入数据,对应公式中的`x`。
  • 实际数据格式由参数data_format决定。
  • NCHW/NHWC:shape支持4D。
  • NCDHW/NDHWC:shape支持5D。
  • ND:shape支持2~8D,其中第2维固定为C轴。
FLOAT32、FLOAT16、BFLOAT16 NCHW/NHWC/NCDHW/NDHWC/ND
gamma 输入
  • 不支持空Tensor。
  • 一个1D张量,表示进行InstanceNorm计算的缩放因子(权重),对应公式中的`γ`。
  • 数据类型默认与x一致;若不一致,则显示设为FLOAT32。
  • shape与输入x的C轴保持一致。
FLOAT32、FLOAT16、BFLOAT16 ND
beta 输入
  • 不支持空Tensor。
  • 一个1D张量,表示进行InstanceNorm计算的偏置,对应公式中的`β`。
  • 数据类型、shape与输入gamma保持一致。
FLOAT32、FLOAT16、BFLOAT16 ND
data_format 可选属性
  • 指定输入x的数据格式。
  • 默认值为NDHWC。
  • 参数data_format仅支持"NCHW"、"NHWC"、"NCDHW"、"NDHWC"、"ND"。
STRING -
epsilon 可选属性
  • 表示添加到方差中的值,以避免出现除以零的情况。对应公式中的`ε`。
  • 默认值为1e-6。
FLOAT32 -
y 输出
  • 支持空Tensor。
  • 表示InstanceNorm的输出结果,对应公式中的`y`。
  • shape、数据类型和数据格式与输入x的保持一致。
FLOAT32、FLOAT16、BFLOAT16 NCHW/NHWC/NCDHW/NDHWC/ND
mean 输出
  • 不支持空Tensor。
  • 表示InstanceNorm的均值,对应公式中的`E(x)`。
  • 数据类型与gamma保持一致。
  • shape与输入x满足broadcast关系(shape维度数与输入x一致,非规约轴shape大小与输入x保持一致,规约轴shape大小为1)。
FLOAT32、FLOAT16、BFLOAT16 ND
variance 输出
  • 不支持空Tensor。
  • 表示InstanceNorm的方差,对应公式中的`Var(x)`。
  • 数据类型、shape与输入mean保持一致。
FLOAT32、FLOAT16、BFLOAT16 ND

约束说明

调用说明

调用方式 调用样例 说明
图模式调用 test_geir_instance_norm 通过算子IR构图方式调用InstanceNorm算子。