DML_ACTIVATION_HARDMAX_OPERATOR_DESC struttura (directml.h)
Esegue una funzione hardmax su ogni elemento di InputTensor, posizionando il risultato nell'elemento corrispondente di OutputTensor.
L'operatore calcola il valore hardmax (1 per la prima occorrenza del valore più grande nel livello e 0 per tutti gli altri valori) di ogni riga nell'input specificato.
Sintassi
struct DML_ACTIVATION_HARDMAX_OPERATOR_DESC {
const DML_TENSOR_DESC *InputTensor;
const DML_TENSOR_DESC *OutputTensor;
};
Members
InputTensor
Tipo: const DML_TENSOR_DESC*
Tensor da cui leggere per l'input. Questo tensore deve avere un rango effettivo non superiore a 2. Il rango effettivo di un tensore è DimensionCount del tensore, escluso le dimensioni più a sinistra delle dimensioni 1. Ad esempio, una dimensione tensor di { 1, 1, BatchCount, Width } è valida ed è equivalente a un tensore di dimensioni { BatchCount, Width }.
OutputTensor
Tipo: const DML_TENSOR_DESC*
Tensore di output in cui scrivere i risultati.
Commenti
L'operatore calcola il valore hardmax (1 per il primo valore massimo e 0 per tutti gli altri) per ogni livello nel batch dell'input specificato. L'input è un tensore 2D (Tensor) di dimensioni (batch_size x input_feature_dimensions). Il tensore di output ha la stessa forma e contiene i valori hardmax dell'input corrispondente.
Disponibilità
Questo operatore è stato introdotto in DML_FEATURE_LEVEL_1_0.
Vincoli tensor
InputTensor e OutputTensor devono avere lo stesso oggetto DataType, DimensionCount e Sizes.
Supporto di Tensor
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 |
DML_FEATURE_LEVEL_1_0 e versioni successive
| Tensore | Tipo | Conteggi delle dimensioni supportate | Tipi di dati supportati |
|---|---|---|---|
| InputTensor | Input | 4 | FLOAT32, FLOAT16 |
| OutputTensor | Output | 4 | FLOAT32, FLOAT16 |
Requisiti
| Requisito | Valore |
|---|---|
| Intestazione | directml.h |