Compartir a través de

estructura DML_ELEMENT_WISE_POW_OPERATOR_DESC (directml.h)

  • Artículo

Calcula cada elemento de InputTensor elevado a la potencia del elemento correspondiente de ExponentTensor, colocando el resultado en el elemento correspondiente de OutputTensor.

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

Las bases negativas se admiten para los exponentes con valores enteros (aunque el tipo de datos todavía puede ser float), de lo contrario, este operador devuelve NaN.

Cuando el tensor de entrada y el tensor exponente tienen un tipo de datos entero, este operador garantiza resultados exactos.

Este operador admite la ejecución en contexto, lo que significa que OutputTensor tiene permiso para alias InputTensor durante el enlace.

Sintaxis

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;
};

Miembros

InputTensor

Tipo: const DML_TENSOR_DESC*

Tensor que contiene los valores de entrada.

ExponentTensor

Tipo: const DML_TENSOR_DESC*

Tensor que contiene los valores del exponente.

OutputTensor

Tipo: const DML_TENSOR_DESC*

Tensor de salida en el que se van a escribir los resultados.

ScaleBias

Tipo: _Maybenull_ const DML_SCALE_BIAS*

Escala y sesgo opcionales que se van a aplicar a la entrada. Si está presente, esto tiene el efecto de aplicar la función g(x) = x * scale + bias a cada elemento de entrada antes de calcular este operador.

Comentarios

Hasta DML_FEATURE_LEVEL_3_0, ExponentTensordebe coincidir con el tipo de InputTensor.

Consulte DML_ELEMENT_WISE_CONSTANT_POW_OPERATOR_DESC para obtener un operador POW que acepte una FLOAT constante para el exponente.

Disponibilidad

Este operador se introdujo en DML_FEATURE_LEVEL_1_0.

Restricciones tensor

  • ExponentTensor, InputTensor y OutputTensor deben tener el mismo DimensionCount y Sizes.
  • InputTensor y OutputTensor deben tener el mismo tipo de datos.

Compatibilidad con Tensor

DML_FEATURE_LEVEL_3_0 y versiones posteriores

Tensor Kind Recuentos de dimensiones admitidos Tipos de datos admitidos
InputTensor Entrada De 1 a 8 FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8
ExponentTensor Entrada De 1 a 8 FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8
OutputTensor Resultados De 1 a 8 FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_1_0 y versiones posteriores

Tensor Kind Recuentos de dimensiones admitidos Tipos de datos admitidos
InputTensor Entrada 4 FLOAT32, FLOAT16
ExponentTensor Entrada 4 FLOAT32, FLOAT16
OutputTensor Resultados 4 FLOAT32, FLOAT16

Requisitos

   
Encabezado directml.h

Consulte también

DML_ELEMENT_WISE_CONSTANT_POW_OPERATOR_DESC