DML_ELEMENT_WISE_ATAN_YX_OPERATOR_DESC結構 (directml.h)
計算 ATensor 和 BTensor 之每個元素的 2 自變數反正切值,其中 ATensor 是 Y 軸 , 而 BTensor 是 X 軸,並將結果放入 OutputTensor 的對應元素。 當 ATensor 和 BTensor 都是對應元素的 0) 時,這個運算符未針對原始 (定義。
f(y, x) = atan2(y, x)
此運算符支援就地執行,這表示輸出張量可在系結期間將 ATensor 或 BTensor 別名化。
語法
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 條件約束
ATensor、 BTensor 和 OutputTensor 必須具有相同 的 DataType、 DimensionCount 和 Sizes。
Tensor 支援
張 | 種類 | 支援的維度計數 | 支援的資料類型 |
---|---|---|---|
ATensor | 輸入 | 1 到 8 | FLOAT32,FLOAT16 |
BTensor | 輸入 | 1 到 8 | FLOAT32,FLOAT16 |
OutputTensor | 輸出 | 1 到 8 | FLOAT32,FLOAT16 |
規格需求
需求 | 值 |
---|---|
最低支援的用戶端 | Windows 組建 22000 |
最低支援的伺服器 | Windows 組建 22000 |
標頭 | directml.h |