Condividi tramite


struttura DML_ELEMENT_WISE_ATAN_YX_OPERATOR_DESC (directml.h)

Calcola l'arcotangente a 2 argomenti per ogni elemento di ATensor e BTensor, dove ATensor è l'asse Y e BTensor è l'asse X, inserendo il risultato nell'elemento corrispondente di OutputTensor. Questo operatore non è definito per l'origine, ovvero quando ATensor e BTensor sono entrambi 0 per gli elementi corrispondenti.

GRU_Forward

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

Questo operatore supporta l'esecuzione sul posto, ovvero il tensore di output è consentito per l'alias ATensor o BTensor durante l'associazione.

Sintassi

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

Members

ATensor

Tipo: const DML_TENSOR_DESC*

Tensore di input da cui leggere i valori dell'asse Y.

BTensor

Tipo: const DML_TENSOR_DESC*

Tensore di input da cui leggere i valori dell'asse X.

OutputTensor

Tipo: const DML_TENSOR_DESC*

Tensore di output in cui scrivere i risultati.

Commenti

Disponibilità

Questo operatore è stato introdotto in DML_FEATURE_LEVEL_3_1.

Vincoli tensor

ATensor, BTensor e OutputTensor devono avere gli stessi oggetti DataType, DimensionCount e Size.

Supporto di Tensor

Tensore Tipo Conteggi delle dimensioni supportati Tipi di dati supportati
ATensor Input da 1 a 8 FLOAT32, FLOAT16
BTensor Input da 1 a 8 FLOAT32, FLOAT16
OutputTensor Output da 1 a 8 FLOAT32, FLOAT16

Requisiti

Requisito Valore
Client minimo supportato Windows Build 22000
Server minimo supportato Windows Build 22000
Intestazione directml.h