共用方式為


DML_ELEMENT_WISE_ATAN_YX_OPERATOR_DESC結構 (directml.h)

計算 ATensorBTensor 之每個元素的 2 自變數反正切值,其中 ATensorY 軸而 BTensorX 軸,並將結果放入 OutputTensor 的對應元素。 當 ATensorBTensor 都是對應元素的 0) 時,這個運算符未針對原始 (定義。

GRU_Forward

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

此運算符支援就地執行,這表示輸出張量可在系結期間將 ATensorBTensor 別名化。

語法

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

成員

ATensor

類型: const DML_TENSOR_DESC*

要從中讀取 Y 軸值的輸入張量。

BTensor

類型: const DML_TENSOR_DESC*

要從中讀取 X 軸值的輸入張量。

OutputTensor

類型: const DML_TENSOR_DESC*

要寫入結果的輸出張量。

備註

可用性

這個運算子是在 中 DML_FEATURE_LEVEL_3_1引進。

Tensor 條件約束

ATensorBTensorOutputTensor 必須具有相同 的 DataTypeDimensionCountSizes

Tensor 支援

種類 支援的維度計數 支援的資料類型
ATensor 輸入 1 到 8 FLOAT32,FLOAT16
BTensor 輸入 1 到 8 FLOAT32,FLOAT16
OutputTensor 輸出 1 到 8 FLOAT32,FLOAT16

規格需求

需求
最低支援的用戶端 Windows 組建 22000
最低支援的伺服器 Windows 組建 22000
標頭 directml.h