estrutura DML_ACTIVATION_HARDMAX_OPERATOR_DESC (directml.h)
Executa uma função hardmax em cada elemento de InputTensor, colocando o resultado no elemento correspondente de OutputTensor.
O operador calcula o hardmax (1 para a primeira ocorrência do maior valor na camada e 0 para todos os outros valores) de cada linha na entrada fornecida.
Sintaxe
struct DML_ACTIVATION_HARDMAX_OPERATOR_DESC {
const DML_TENSOR_DESC *InputTensor;
const DML_TENSOR_DESC *OutputTensor;
};
Membros
InputTensor
Tipo: const DML_TENSOR_DESC*
O tensor do qual ler para a entrada. 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.
Comentários
O operador calcula os valores hardmax (1 para o primeiro valor máximo e 0 para todos os outros) para cada camada no lote da entrada fornecida. A entrada é um tensor 2D (Tensor) de tamanho (batch_size x input_feature_dimensions). O tensor de saída tem a mesma forma e contém os valores hardmax da entrada correspondente.
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 |