estrutura DML_ACTIVATION_LINEAR_OPERATOR_DESC (directml.h)

Artigo
02/29/2024

Executa a função de ativação linear em cada elemento em InputTensor, colocando o resultado no elemento correspondente de OutputTensor.

f(x) = Alpha * x + Beta.

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_LINEAR_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  FLOAT                 Alpha;
  FLOAT                 Beta;
};

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 no qual gravar os resultados.

Alpha

Tipo: FLOAT

O coeficiente alfa.

Beta

Tipo: FLOAT

O coeficiente beta.

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

Compartilhar via