Condividi tramite

struttura DML_ACTIVATION_PARAMETERIZED_RELU_OPERATOR_DESC (directml.h)

  • Articolo

Esegue una funzione di attivazione rectified linear unit (ReLU) con parametri su ogni elemento di InputTensor, inserendo il risultato nell'elemento corrispondente di OutputTensor.

f(x, slope) = x,         if x >= 0
              slope * x, otherwise

Questo operatore supporta l'esecuzione sul posto, vale a dire che il tensore di output è autorizzato ad alias InputTensor durante l'associazione.

Sintassi

struct DML_ACTIVATION_PARAMETERIZED_RELU_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *SlopeTensor;
  const DML_TENSOR_DESC *OutputTensor;
};

Members

InputTensor

Tipo: const DML_TENSOR_DESC*

Tensore di input da cui leggere.

SlopeTensor

Tipo: const DML_TENSOR_DESC*

Tensore contenente il coefficiente angolare per ogni valore corrispondente dell'input.

OutputTensor

Tipo: const DML_TENSOR_DESC*

Tensore di output in cui scrivere i risultati.

Requisiti

   
Intestazione directml.h

Vedi anche

DML_ACTIVATION_RELU_OPERATOR_DESC

Disponibilità

Questo operatore è stato introdotto in DML_FEATURE_LEVEL_1_0.

Vincoli tensor

InputTensor, OutputTensor e SlopeTensor devono avere gli stessi Valori DataType, DimensionCount e Size.

Supporto di Tensor

DML_FEATURE_LEVEL_5_1 e versioni successive

Tensore Tipo Conteggi delle dimensioni supportati Tipi di dati supportati
InputTensor Input da 1 a 8 FLOAT32, FLOAT16, INT32, INT16, INT8
SlopeTensor Input da 1 a 8 FLOAT32, FLOAT16, INT32, INT16, INT8
OutputTensor Output da 1 a 8 FLOAT32, FLOAT16, INT32, INT16, INT8

DML_FEATURE_LEVEL_3_0 e versioni successive

Tensore Tipo Conteggi delle dimensioni supportati Tipi di dati supportati
InputTensor Input da 1 a 8 FLOAT32, FLOAT16
SlopeTensor Input da 1 a 8 FLOAT32, FLOAT16
OutputTensor Output da 1 a 8 FLOAT32, FLOAT16

DML_FEATURE_LEVEL_2_0 e versioni successive

Tensore Tipo Conteggi delle dimensioni supportati Tipi di dati supportati
InputTensor Input da 4 a 5 FLOAT32, FLOAT16
SlopeTensor Input da 4 a 5 FLOAT32, FLOAT16
OutputTensor Output da 4 a 5 FLOAT32, FLOAT16

DML_FEATURE_LEVEL_1_0 e versioni successive

Tensore Tipo Conteggi delle dimensioni supportati Tipi di dati supportati
InputTensor Input 4 FLOAT32, FLOAT16
SlopeTensor Input 4 FLOAT32, FLOAT16
OutputTensor Output 4 FLOAT32, FLOAT16