estrutura DML_ACTIVATION_SHRINK_OPERATOR_DESC (directml.h)
Executa a função de ativação de redução em cada elemento em InputTensor, colocando o resultado no elemento correspondente de OutputTensor.
f(x) = x - Bias, if x > Threshold
x + Bias, if x < -Threshold
0, otherwise
Esse operador dá suporte à execução in-loco, o que significa que o tensor de saída tem permissão para alias InputTensor durante a associação.
Sintaxe
struct DML_ACTIVATION_SHRINK_OPERATOR_DESC {
const DML_TENSOR_DESC *InputTensor;
const DML_TENSOR_DESC *OutputTensor;
FLOAT Bias;
FLOAT Threshold;
};
Membros
InputTensor
Tipo: const DML_TENSOR_DESC*
O tensor de entrada do qual ler.
OutputTensor
Tipo: const DML_TENSOR_DESC*
O tensor de saída para o qual gravar os resultados.
Bias
Tipo: FLOAT
O valor do viés. Um padrão típico para esse valor é 0,0.
Threshold
Tipo: FLOAT
O valor do limite. Um padrão típico para esse valor é 0,5.
Comentários
Disponibilidade
Esse operador foi introduzido em DML_FEATURE_LEVEL_2_0
.
Restrições do Tensor
InputTensor e OutputTensor devem ter os mesmos DataType, DimensionCount e Sizes.
Suporte ao Tensor
DML_FEATURE_LEVEL_5_1 e superior
Tensor | Tipo | Contagens de dimensões com suporte | Tipos de dados com suporte |
---|---|---|---|
InputTensor | Entrada | 1 a 8 | FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8 |
OutputTensor | Saída | 1 a 8 | FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8 |
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 |
Requisitos
Requisito | Valor |
---|---|
Cliente mínimo com suporte | Windows 10, versão 2004 (10.0; Build 19041) |
Servidor mínimo com suporte | Windows Server, versão 2004 (10.0; Build 19041) |
Cabeçalho | directml.h |