DML_QUANTIZED_LINEAR_AVERAGE_POOLING_OPERATOR_DESC 構造体 (directml.h)

[アーティクル]
03/16/2024

入力テンソルのスライディングウィンドウ内の要素の量子化された値を平均化します。この演算子は、数学的には、入力をデカンタイズしてから平均プーリングを実行し、出力を量子化することと同じです。

デカンタイズ関数

f(Input, Scale, ZeroPoint) = (Input - ZeroPoint) * Scale

量子化関数

f(Input, Scale, ZeroPoint) = clamp(round(Input / Scale) + ZeroPoint, Min, Max)

重要

この API は、DirectML スタンドアロン再頒布可能パッケージの一部として使用できます (Microsoft.AI.DirectML バージョン 1.13 以降を参照してください)。 DirectML バージョン履歴も参照してください。

構文

struct DML_QUANTIZED_LINEAR_AVERAGE_POOLING_OPERATOR_DESC
{
    const DML_TENSOR_DESC* InputTensor;
    const DML_TENSOR_DESC* InputScaleTensor;
    _Maybenull_ const DML_TENSOR_DESC* InputZeroPointTensor;
    const DML_TENSOR_DESC* OutputScaleTensor;
    _Maybenull_ const DML_TENSOR_DESC* OutputZeroPointTensor;
    const DML_TENSOR_DESC* OutputTensor;
    UINT DimensionCount;
    _Field_size_(DimensionCount) const UINT* Strides;
    _Field_size_(DimensionCount) const UINT* WindowSize;
    _Field_size_(DimensionCount) const UINT* StartPadding;
    _Field_size_(DimensionCount) const UINT* EndPadding;
    _Field_size_(DimensionCount) const UINT* Dilations;
    BOOL IncludePadding;
};

メンバー

InputTensor

型: const DML_TENSOR_DESC*

4D には { BatchCount, ChannelCount, Height, Width }、5D には { BatchCount, ChannelCount, Depth, Height, Weight } のサイズの入力テンソル。

InputScaleTensor

型: const DML_TENSOR_DESC*

InputTensor スケールデータを含むテンソル。 InputScaleTensor の予想される次元は、テンソル単位の量子化が必要な場合は { 1, 1, 1, 1 }、チャネル単位の量子化が必要な場合は { 1, ChannelCount, 1, 1 } です。これらのスケール値は、InputTensor 値をデカンタイズするために使用されます。

InputZeroPointTensor

型: _Maybenull_ const DML_TENSOR_DESC*

InputTensor ゼロポイントデータを含む省略可能なテンソル。 InputZeroPointTensor の予想される次元は、テンソル単位の量子化が必要な場合は { 1, 1, 1, 1 }、チャネル単位の量子化が必要な { 1, ChannelCount, 1, 1 } 場合です。これらのゼロポイント値は、InputTensor 値をデカンタイズするために使用されます。

OutputScaleTensor

型: const DML_TENSOR_DESC*

OutputTensor スケールデータを含むテンソル。 OutputScaleTensor の予想される次元は、テンソル単位の量子化が必要な場合は { 1, 1, 1, 1 }、チャネルごとの量子化が必要な場合は { 1, ChannelCount, 1, 1 } です。これらのスケール値は、OutputTensor 値を量子化するために使用されます。

OutputZeroPointTensor

型: _Maybenull_ const DML_TENSOR_DESC*

OutputTensor のゼロポイントデータを含む省略可能なテンソル。 OutputZeroPointTensor の予想される次元は、テンソル単位の量子化が必要な場合は { 1, 1, 1, 1 }、チャネル単位の量子化が必要な場合は { 1, ChannelCount, 1, 1 } です。このゼロポイント値は、OutputTensor 値を量子化するために使用されます。

OutputTensor

型: const DML_TENSOR_DESC*

出力テンソルの説明。出力テンソルのサイズは次のように計算できます。

OutputTensor->Sizes[0] = InputTensor->Sizes[0];
OutputTensor->Sizes[1] = InputTensor->Sizes[1];

for (UINT i = 0; i < DimensionCount; ++i) {
  UINT PaddedSize = InputTensor->Sizes[i + 2] + StartPadding[i] + EndPadding[i];
  OutputTensor->Sizes[i + 2] = (PaddedSize - WindowSizes[i]) / Strides[i] + 1;
}

DimensionCount

型: UINT

入力テンソル InputTensor の空間次元の数。これは、スライディングウィンドウ WindowSize の次元の数にも対応します。この値は、Strides、StartPadding、ndPadding 配列のサイズも決定します。 InputTensor が 4D の場合は 2、5D テンソルの場合は 3 に設定する必要があります。

Strides