Condividi tramite


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

Vedi anche

struttura DML_ACTIVATION_SOFTMAX_OPERATOR_DESC

struttura DML_ARGMAX_OPERATOR_DESC