共用方式為

DML_MEAN_VARIANCE_NORMALIZATION2_OPERATOR_DESC結構 （directml.h）

待定

這很重要

此 API 可作為 DirectML 獨立可轉散發套件的一部分（請參閱 Microsoft.AI.DirectML 1.15.0 版和更新版本。 另請參閱 DirectML 版本歷程記錄。

語法

struct DML_MEAN_VARIANCE_NORMALIZATION2_OPERATOR_DESC
{
    const DML_TENSOR_DESC* InputTensor;
    _Maybenull_ const DML_TENSOR_DESC* ScaleTensor;
    _Maybenull_ const DML_TENSOR_DESC* BiasTensor;
    const DML_TENSOR_DESC* OutputTensor;
    UINT AxisCount;
    _Field_size_(AxisCount) const UINT* Axes;
    BOOL UseMean;
    BOOL UseVariance;
    FLOAT Epsilon;
    _Maybenull_ const DML_OPERATOR_DESC* FusedActivation;
};

成員

InputTensor

類型： const DML_TENSOR_DESC*

包含輸入數據的張量。 這個張量維度應該是 { BatchCount, ChannelCount, Height, Width }。

ScaleTensor

類型： _Maybenull_ const DML_TENSOR_DESC*

包含 Scale 資料的選擇性張量。

如果 DML_FEATURE_LEVEL 小於 DML_FEATURE_LEVEL_4_0，則這個張量維度應該是 { ScaleBatchCount, ChannelCount, ScaleHeight, ScaleWidth }。 ScaleBatchCount、ScaleHeight 和 ScaleWidth 維度應該符合 InputTensor，或設定為 1，以自動在輸入之間廣播這些維度。

如果 DML_FEATURE_LEVEL 大於或等於 DML_FEATURE_LEVEL_4_0，則任何維度都可以設定為 1，並自動廣播以符合 InputTensor。

如果 DML_FEATURE_LEVEL 小於 DML_FEATURE_LEVEL_5_2，則 如果 BiasTensor 存在，則需要此張量。 如果 DML_FEATURE_LEVEL 大於或等於 DML_FEATURE_LEVEL_5_2，則不論 BiasTensor 的值為何，此張量都可以為 null。

BiasTensor

類型： _Maybenull_ const DML_TENSOR_DESC*

包含偏差數據的選擇性張量。

如果 DML_FEATURE_LEVEL 小於 DML_FEATURE_LEVEL_4_0，則這個張量維度應該是 { BiasBatchCount, ChannelCount, BiasHeight, BiasWidth }。 Dimensions BiasBatchCount、BiasHeight 和 BiasWidth 應該符合 InputTensor，或設定為 1，以自動在輸入之間廣播這些維度。

如果 DML_FEATURE_LEVEL 大於或等於 DML_FEATURE_LEVEL_4_0，則任何維度都可以設定為 1，並自動廣播以符合 InputTensor。

如果 DML_FEATURE_LEVEL 小於 DML_FEATURE_LEVEL_5_2，則如果 ScaleTensor 存在，則需要此張量。 如果 DML_FEATURE_LEVEL 大於或等於 DML_FEATURE_LEVEL_5_2，則不論 ScaleTensor 的值為何，此張量都可以是 null。

OutputTensor

類型： const DML_TENSOR_DESC*

要寫入結果的張量。 此張量維度為 { BatchCount, ChannelCount, Height, Width }。

AxisCount

類型： UINT

座標軸的數目。 此欄位會決定 Axes 陣列的大小。

Axes

類型：_Field_size_（AxisCount） const UINT*

要計算平均值和變異數的座標軸。

UseMean

類型： BOOL

待定

UseVariance

類型： BOOL

待定

Epsilon

類型： FLOAT

要用來避免除以零的 epsilon 值。 建議使用0.00001作為預設值。

FusedActivation

類型： _Maybenull_ const DML_OPERATOR_DESC*

在正規化之後要套用的選擇性融合啟用層。

可用性

此運算子是在 DML_FEATURE_LEVEL_6_3 中引進的。

Tensor 條件約束

BiasTensor、InputTensor、OutputTensor 和 ScaleTensor 必須具有相同的 DataType 和 DimensionCount。

Tensor 支援

張量 仁慈 (if context refers to kindness) 支援的維度計數 支援的數據類型
InputTensor 的 輸入 1 到 8 FLOAT32，FLOAT16
ScaleTensor 的 選擇性輸入 1 到 8 FLOAT32，FLOAT16
BiasTensor 選擇性輸入 1 到 8 FLOAT32，FLOAT16
OutputTensor 輸出 1 到 8 FLOAT32，FLOAT16
  • Last updated on