次の方法で共有


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_LEVELDML_FEATURE_LEVEL_4_0未満の場合は、このテンソルのディメンションを{ ScaleBatchCount, ChannelCount, ScaleHeight, ScaleWidth }する必要があります。 Dimensions ScaleBatchCount、ScaleHeight、ScaleWidth は InputTensor と一致するか、1 に設定して入力全体でそれらのディメンションを自動的にブロードキャストする必要があります。

DML_FEATURE_LEVELDML_FEATURE_LEVEL_4_0以上の場合は、任意のディメンションを 1 に設定し、InputTensor と一致するように自動的にブロードキャストできます。

DML_FEATURE_LEVELDML_FEATURE_LEVEL_5_2未満の場合、BiasTensor が存在する場合は、このテンソルが必要です。 DML_FEATURE_LEVELDML_FEATURE_LEVEL_5_2以上の場合、BiasTensor の値に関係なく、このテンソルを null にすることができます。

BiasTensor

型: _Maybenull_ const DML_TENSOR_DESC*

Bias データを含む省略可能なテンソル。

DML_FEATURE_LEVELDML_FEATURE_LEVEL_4_0未満の場合は、このテンソルのディメンションを{ BiasBatchCount, ChannelCount, BiasHeight, BiasWidth }する必要があります。 Dimensions BiasBatchCount、BiasHeight、BiasWidth は InputTensor と一致するか、1 に設定して入力全体でそれらのディメンションを自動的にブロードキャストする必要があります。

DML_FEATURE_LEVELDML_FEATURE_LEVEL_4_0以上の場合は、任意のディメンションを 1 に設定し、InputTensor と一致するように自動的にブロードキャストできます。

DML_FEATURE_LEVELDML_FEATURE_LEVEL_5_2未満の場合、ScaleTensor が存在する場合は、このテンソルが必要です。 DML_FEATURE_LEVELDML_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 制約

BiasTensorInputTensorOutputTensorScaleTensor は、 DataTypeDimensionCount が同じである必要があります。

Tensor のサポート

テンソル 優しい サポートされているディメンション数 サポートされているデータ型
インプットテンソル インプット 1 から 8 FLOAT32、FLOAT16
スケールテンソル 省略可能な入力 1 から 8 FLOAT32、FLOAT16
バイアステンソル 省略可能な入力 1 から 8 FLOAT32、FLOAT16
出力テンソル アウトプット 1 から 8 FLOAT32、FLOAT16