DML_ACTIVATION_SHRINK_OPERATOR_DESC struttura (directml.h)

  • Articolo

Esegue la funzione di attivazione della compattazione su ogni elemento in InputTensor, posizionando il risultato nell'elemento corrispondente di OutputTensor.

f(x) = x - Bias, if x > Threshold
       x + Bias, if x < -Threshold
       0,        otherwise

Questo operatore supporta l'esecuzione sul posto, ovvero il tensore di output è consentito per alias InputTensor durante l'associazione.

Sintassi

struct DML_ACTIVATION_SHRINK_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  FLOAT                 Bias;
  FLOAT                 Threshold;
};

Members

InputTensor

Tipo: const DML_TENSOR_DESC*

Tensore di input da cui leggere.

OutputTensor

Tipo: const DML_TENSOR_DESC*

Tensore di output in cui scrivere i risultati.

Bias

Tipo: FLOAT

Valore della distorsione. Un valore predefinito tipico per questo valore è 0,0.

Threshold

Tipo: FLOAT

Valore della soglia. Un valore predefinito tipico per questo valore è 0,5.

Commenti

Disponibilità

Questo operatore è stato introdotto in DML_FEATURE_LEVEL_2_0.

Vincoli tensor

InputTensor e OutputTensor devono avere lo stesso oggetto DataType, DimensionCount e Sizes.

Supporto di Tensor

DML_FEATURE_LEVEL_5_1 e versioni successive

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

DML_FEATURE_LEVEL_3_0 e versioni successive

Tensore Tipo Conteggi delle dimensioni supportate Tipi di dati supportati
InputTensor 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 supportate Tipi di dati supportati
InputTensor Input da 4 a 5 FLOAT32, FLOAT16
OutputTensor Output da 4 a 5 FLOAT32, FLOAT16

Requisiti

Requisito Valore
Client minimo supportato Windows 10 versione 2004 (10.0; Build 19041)
Server minimo supportato Windows Server, versione 2004 (10.0; Build 19041)
Intestazione directml.h