estrutura DML_ACTIVATION_SOFTMAX_OPERATOR_DESC (directml.h)

Executa uma função de ativação softmax em InputTensor, colocando o resultado no elemento correspondente de 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]))

Onde exp(x) é a função de exponenciação natural.

Sintaxe

struct DML_ACTIVATION_SOFTMAX_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
};

Membros

InputTensor

Tipo: const DML_TENSOR_DESC*

O tensor de entrada do qual ler. Esse tensor deve ter uma classificação efetiva não maior que 2. A classificação efetiva de um tensor é a DimensionCount do tensor, excluindo as dimensões mais à esquerda do tamanho 1. Por exemplo, um tamanho tensor de { 1, 1, BatchCount, Width } é válido e é equivalente a um tensor de tamanhos { BatchCount, Width }.

OutputTensor

Tipo: const DML_TENSOR_DESC*

O tensor de saída no qual gravar os resultados.

Disponibilidade

Esse operador foi introduzido no DML_FEATURE_LEVEL_1_0.

Restrições do Tensor

InputTensor e OutputTensor devem ter os mesmos DataType, DimensionCount e Sizes.

Suporte ao Tensor

DML_FEATURE_LEVEL_3_0 e superior

Tensor	Tipo	Contagens de dimensões com suporte	Tipos de dados com suporte
InputTensor	Entrada	1 a 8	FLOAT32, FLOAT16
OutputTensor	Saída	1 a 8	FLOAT32, FLOAT16

DML_FEATURE_LEVEL_2_0 e superior

Tensor	Tipo	Contagens de dimensões com suporte	Tipos de dados com suporte
InputTensor	Entrada	4 a 5	FLOAT32, FLOAT16
OutputTensor	Saída	4 a 5	FLOAT32, FLOAT16

DML_FEATURE_LEVEL_1_0 e superior

Tensor	Tipo	Contagens de dimensões com suporte	Tipos de dados com suporte
InputTensor	Entrada	4	FLOAT32, FLOAT16
OutputTensor	Saída	4	FLOAT32, FLOAT16

Requisitos

Requisito	Valor
Cabeçalho	directml.h

Confira também

estrutura DML_ACTIVATION_LOG_SOFTMAX_OPERATOR_DESC

estrutura DML_ARGMAX_OPERATOR_DESC