estrutura DML_ELEMENT_WISE_RECIP_OPERATOR_DESC (directml.h)

Artigo
08/23/2023

Calcula a recíproca para cada elemento do tensor de entrada, colocando o resultado no elemento correspondente do tensor de saída.

f(x) = 1 / x

Esse operador dá suporte à execução in-loco, o que significa que o tensor de saída tem permissão para alias do tensor de entrada durante a associação.

Sintaxe

struct DML_ELEMENT_WISE_RECIP_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  const DML_SCALE_BIAS  *ScaleBias;
};

Membros

InputTensor

Tipo: const DML_TENSOR_DESC*

O tensor do qual ler para o primeiro tensor de entrada, x.

OutputTensor

Tipo: const DML_TENSOR_DESC*

O tensor de saída para o qual gravar os resultados.

ScaleBias

Tipo: _Maybenull_ const DML_SCALE_BIAS*

Uma escala e um viés opcionais a serem aplicados à entrada. Se estiver presente, isso terá o efeito de aplicar a função g(x) = x * scale + bias a cada elemento de entrada antes de calcular esse operador.

Disponibilidade

Esse operador foi introduzido em 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_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


Cabeçalho	directml.h

Compartilhar via