다음을 통해 공유


DML_AVERAGE_POOLING1_OPERATOR_DESC 구조체(directml.h)

입력 텐서의 슬라이딩 윈도우 내 요소에 대한 값의 평균을 계산합니다.

중요합니다

이 API는 DirectML 독립 실행형 재배포 가능 패키지의 일부로 사용할 수 있습니다( Microsoft.AI.DirectML 버전 1.13 이상 참조). 또한 DirectML 버전 기록 참조하세요.

문법

struct DML_AVERAGE_POOLING1_OPERATOR_DESC
{
    const DML_TENSOR_DESC* InputTensor;
    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 및 5D에 대한 { BatchCount, ChannelCount, Height, Width }{ BatchCount, ChannelCount, Depth, Height, Weight } 입력 텐서입니다.

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

슬라이딩 윈도우 WindowSize의 차원 수에 해당하는 입력 텐서 InputTensor의 공간 차원 수입니다. 이 값은 Strides, StartPaddingEndPadding 배열의 크기도 결정합니다. InputTensor가 4D이면 2로, 5D 텐서인 경우 3으로 설정해야 합니다.

Strides

형식: _Field_size_(DimensionCount) const UINT*

{ Height, Width }가 2로 설정되거나 3으로 설정된 경우 크기의 { Depth, Height, Width } 슬라이딩 윈도우 크기에 대한 보폭입니다.

WindowSize

형식: _Field_size_(DimensionCount) const UINT*

{ Height, Width }가 2로 설정되거나 { Depth, Height, Width } 3으로 설정된 경우의 슬라이딩 윈도우 크기입니다.

StartPadding

형식: _Field_size_(DimensionCount) const UINT*

입력 텐서 InputTensor의 각 공간 차원의 시작 부분에 적용할 패딩 요소의 수입니다. 값은 { Height, Width }DimensionCount 가 2로 설정되거나 { Depth, Height, Width } 3으로 설정된 경우에 있습니다.

EndPadding

형식: _Field_size_(DimensionCount) const UINT*

입력 텐서 InputTensor의 각 공간 차원 끝에 적용할 패딩 요소의 수입니다. 값은 { Height, Width }DimensionCount 가 2로 설정되거나 { Depth, Height, Width } 3으로 설정된 경우에 있습니다.

Dilations

형식: _Field_size_(DimensionCount) const UINT*

해당 값의 모든 요소에 대해 슬라이딩 윈도우 내의 요소가 선택된 입력 텐 서 InputTensor 의 각 공간 차원에 대한 값입니다. 값은 { Height, Width }DimensionCount 가 2로 설정되거나 { Depth, Height, Width } 3으로 설정된 경우에 있습니다.

IncludePadding

형식: BOOL

슬라이딩 윈도우 내의 모든 요소에서 평균 값을 계산할 때 공간 가장자리 주위에 안쪽 여백 요소를 포함할지 여부를 나타냅니다. 값이 FALSE로 설정되면 패딩 요소는 평균 계산의 제수 값의 일부로 계산되지 않습니다.

비고

DML_AVERAGE_POOLING1_OPERATOR_DESC 추가 상수 배열 팽창을 제외하고 DML_AVERAGE_POOLING_OPERATOR_DESC같습니다. 4D 입력의 경우 { 1,1}, 5D 입력 기능의 경우 { 1,1,1}으로 설정된 경우 DML_AVERAGE_POOLING1_OPERATOR_DESCDML_AVERAGE_POOLING_OPERATOR_DESC 동일시됩니다.

가용도

이 연산자는 DML_FEATURE_LEVEL_6_2 도입되었습니다.

Tensor 제약 조건

InputTensorOutputTensor 에는 동일한 DataTypeDimensionCount가 있어야 합니다.

Tensor 지원

텐서 친절한 지원되는 차원 수 지원되는 데이터 형식
입력 텐서 입력 4~5 FLOAT32, FLOAT16
출력텐서 출력 4~5 FLOAT32, FLOAT16

요구 사항

   
머리글 directml.h