Compartilhar via

estrutura DML_ELEMENT_WISE_POW_OPERATOR_DESC (directml.h)

  • Artigo

Calcula cada elemento de InputTensor elevado à potência do elemento correspondente de ExponentTensor, colocando o resultado no elemento correspondente de OutputTensor.

f(input, exponent) = pow(input, exponent)

Há suporte para bases negativas para expoentes com valores integrais (embora o tipo de dados ainda possa ser float), caso contrário, esse operador retornará NaN.

Quando o tensor de entrada e o tensor expoente têm tipo de dados integral, esse operador garante resultados exatos.

Esse operador dá suporte à execução in-loco, o que significa que OutputTensor tem permissão para alias InputTensor durante a associação.

Sintaxe

struct DML_ELEMENT_WISE_POW_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *ExponentTensor;
  const DML_TENSOR_DESC *OutputTensor;
  const DML_SCALE_BIAS  *ScaleBias;
};

Membros

InputTensor

Tipo: const DML_TENSOR_DESC*

O tensor que contém os valores de entrada.

ExponentTensor

Tipo: const DML_TENSOR_DESC*

O tensor que contém os valores do expoente.

OutputTensor

Tipo: const DML_TENSOR_DESC*

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

ScaleBias

Tipo: _Maybenull_ const DML_SCALE_BIAS*

Uma escala e um desvio opcionais a serem aplicados à entrada. Se 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.

Comentários

Até DML_FEATURE_LEVEL_3_0, ExponentTensordeve corresponder ao tipo de InputTensor.

Consulte DML_ELEMENT_WISE_CONSTANT_POW_OPERATOR_DESC para um operador POW que aceita uma FLOAT constante para o expoente.

Disponibilidade

Esse operador foi introduzido no DML_FEATURE_LEVEL_1_0.

Restrições do Tensor

  • ExponentTensor, InputTensor e OutputTensor devem ter os mesmos DimensionCount e Sizes.
  • InputTensor e OutputTensor devem ter o mesmo DataType.

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, INT32, INT16, INT8, UINT32, UINT16, UINT8
ExponentTensor 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_1_0 e superior

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

Requisitos

   
Cabeçalho directml.h

Confira também

DML_ELEMENT_WISE_CONSTANT_POW_OPERATOR_DESC