Compartir a través de

estructura DML_ELEMENT_WISE_ATAN_YX_OPERATOR_DESC (directml.h)

  • Artículo

Calcula la arcotangente de 2 argumentos para cada elemento de ATensor y BTensor, donde ATensor es el eje Y y BTensor es el eje X, colocando el resultado en el elemento correspondiente de OutputTensor. Este operador no está definido para el origen (es decir, cuando ATensor y BTensor son 0 para los elementos correspondientes).

GRU_Forward

f(y, x) = atan2(y, x)

Este operador admite la ejecución local, lo que significa que el tensor de salida puede alias ATensor o BTensor durante el enlace.

Sintaxis

struct DML_ELEMENT_WISE_ATAN_YX_OPERATOR_DESC {
  const DML_TENSOR_DESC *ATensor;
  const DML_TENSOR_DESC *BTensor;
  const DML_TENSOR_DESC *OutputTensor;
};

Miembros

ATensor

Tipo: const DML_TENSOR_DESC*

Tensor de entrada del que se van a leer los valores del eje Y.

BTensor

Tipo: const DML_TENSOR_DESC*

Tensor de entrada del que se van a leer los valores del eje X.

OutputTensor

Tipo: const DML_TENSOR_DESC*

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

Comentarios

Disponibilidad

Este operador se introdujo en DML_FEATURE_LEVEL_3_1.

Restricciones tensor

ATensor, BTensor y OutputTensor deben tener el mismo DataType, DimensionCount y Sizes.

Compatibilidad con Tensor

Tensor Clase Recuentos de dimensiones admitidos Tipos de datos admitidos
ATensor Entrada De 1 a 8 FLOAT32, FLOAT16
BTensor Entrada De 1 a 8 FLOAT32, FLOAT16
OutputTensor Resultados De 1 a 8 FLOAT32, FLOAT16

Requisitos

Requisito Value
Cliente mínimo compatible Windows Build 22000
Servidor mínimo compatible Windows Build 22000
Encabezado directml.h