struttura DML_ACTIVATION_LOG_SOFTMAX_OPERATOR_DESC (directml.h)
Esegue una funzione di attivazione log-of-softmax naturale su ogni elemento di InputTensor, inserendo il risultato nell'elemento corrispondente di OutputTensor.
// Let x_i be the current value in the axis, and j be the total number of elements along that axis.
f(x_i) = ln(exp(x_i) / sum(exp(x_0), ..., exp(x_j)))
Dove exp(x) è la funzione di esponente naturale e ln(x) è il logaritmo naturale.
Sintassi
struct DML_ACTIVATION_LOG_SOFTMAX_OPERATOR_DESC {
const DML_TENSOR_DESC *InputTensor;
const DML_TENSOR_DESC *OutputTensor;
};
Members
InputTensor
Tipo: const DML_TENSOR_DESC*
Tensore di input da cui leggere. Questo tensore deve avere un rango effettivo non maggiore di 2. Il rango effettivo di un tensore è DimensionCount del tensore, escluse 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.
Disponibilità
Questo operatore è stato introdotto in DML_FEATURE_LEVEL_1_0
.
Vincoli tensor
InputTensor e OutputTensor devono avere gli stessi Valori DataType, DimensionCount e Size.
Supporto di Tensor
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 |
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 |
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 |
OutputTensor | Output | 4 | FLOAT32, FLOAT16 |
Requisiti
Intestazione | directml.h |