Condividi tramite

struttura DML_MAX_POOLING_OPERATOR_DESC (directml.h)

  • Articolo

Calcola il valore massimo tra gli elementi all'interno della finestra scorrevole sul tensore di input.

Sintassi

struct DML_MAX_POOLING_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  UINT                  DimensionCount;
  const UINT            *Strides;
  const UINT            *WindowSize;
  const UINT            *StartPadding;
  const UINT            *EndPadding;
};

Membri

InputTensor

Tipo: const DML_TENSOR_DESC*

Un tensore di input di dimensioni { BatchCount, ChannelCount, Height, Width } se InputTensor.DimensionCount è 4 e { BatchCount, ChannelCount, Depth, Height, Weight } se InputTensor.DimensionCount è 5.

OutputTensor

Tipo: const DML_TENSOR_DESC*

Tensore di output in cui scrivere i risultati. Le dimensioni del tensore di output possono essere calcolate nel modo seguente.

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

Tipo: UINT

Numero di dimensioni spaziali del tensore di input InputTensor, che corrisponde anche al numero di dimensioni della finestra scorrevole WindowSize. Questo valore determina anche le dimensioni delleStrides, StartPaddinge matrici di EndPadding. Deve essere impostato su 2 quando InputTensor è 4D e 3 quando si tratta di un tensore 5D.

Strides

Tipo: const UINT*

I passi per le dimensioni della finestra temporale scorrevole delle dimensioni quando la DimensionCount è impostata su 2 o se impostata su 3.

WindowSize

Tipo: _Field_size_(DimensionCount) const UINT*

Dimensioni della finestra temporale scorrevole in { Height, Width } quando DimensionCount è impostato su 2 o { Depth, Height, Width } se impostato su 3.

StartPadding

Tipo: _Field_size_(DimensionCount) const UINT*

Numero di elementi di riempimento da applicare all'inizio di ogni dimensione spaziale del tensore di input InputTensor. I valori si trovano in { Height, Width } quando DimensionCount è impostato su 2 o { Depth, Height, Width } se impostato su 3.

EndPadding

Tipo: _Field_size_(DimensionCount) const UINT*

Numero di elementi di riempimento da applicare alla fine di ogni dimensione spaziale del tensore di input InputTensor. I valori si trovano in { Height, Width } quando DimensionCount è impostato su 2 o { Depth, Height, Width } se impostato su 3.

Osservazioni

Una versione più recente di questo operatore, DML_MAX_POOLING1_OPERATOR_DESC, è stata introdotta in DML_FEATURE_LEVEL_2_0.

Disponibilità

Questo operatore è stato introdotto in DML_FEATURE_LEVEL_1_0.

Vincoli tensor

InputTensor e OutputTensor devono avere lo stesso DataType e DimensionCount.

Supporto tensor

DML_FEATURE_LEVEL_5_0 e versioni successive

Tensore Gentile Conteggi delle dimensioni supportati Tipi di dati supportati
InputTensor Immissione Da 4 a 5 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8
OutputTensor Prodotto Da 4 a 5 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_3_0 e versioni successive

Tensore Gentile Conteggi delle dimensioni supportati Tipi di dati supportati
InputTensor Immissione Da 4 a 5 FLOAT32, FLOAT16, INT8, UINT8
OutputTensor Prodotto Da 4 a 5 FLOAT32, FLOAT16, INT8, UINT8

DML_FEATURE_LEVEL_1_0 e versioni successive

Tensore Gentile Conteggi delle dimensioni supportati Tipi di dati supportati
InputTensor Immissione Da 4 a 5 FLOAT32, FLOAT16
OutputTensor Prodotto Da 4 a 5 FLOAT32, FLOAT16

Fabbisogno

Requisito Valore
intestazione directml.h