DML_ACTIVATION_SOFTMAX_OPERATOR_DESC struttura (directml.h)

Esegue una funzione di attivazione softmax in InputTensor, posizionando il risultato nell'elemento corrispondente di OutputTensor.

For 1-D InputTensor:
// Let x[i] be the current element in the InputTensor, and j be the total number of elements in the InputTensor
f(x[i]) = exp(x[i]) / sum(exp(x[0]), ..., exp(x[j-1]))

Dove exp(x) è la funzione di exponentiation naturale.

Sintassi

struct DML_ACTIVATION_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 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.

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

Vedi anche

DML_ACTIVATION_LOG_SOFTMAX_OPERATOR_DESC struttura

DML_ARGMAX_OPERATOR_DESC struttura