DML_ELEMENT_WISE_ATAN_OPERATOR_DESC 結構 (directml.h)

發行項
02/26/2024

計算 InputTensor 每個元素的反正切值，將結果放入 OutputTensor 的對應元素中。

f(x) = atan(x)

此運算符支援就地執行，這表示 OutputTensor 可在系結期間將 InputTensor 別名化。

語法

struct DML_ELEMENT_WISE_ATAN_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  const DML_SCALE_BIAS  *ScaleBias;
};

成員

InputTensor

類型： const DML_TENSOR_DESC*

要從中讀取的輸入張量。

OutputTensor

類型： const DML_TENSOR_DESC*

要寫入結果的輸出張量。

ScaleBias

類型：_Maybenull_ const DML_SCALE_BIAS*

要套用至輸入的選擇性縮放比例和偏差。如果存在，這會在計算這個運算符之前，將函 g(x) = x * scale + bias 式套用至每個輸入元素的效果。

可用性

這個運算子是在中 DML_FEATURE_LEVEL_1_0引進的。

Tensor 條件約束

InputTensor 和 OutputTensor 必須具有相同 的 DataType、 DimensionCount 和大小。

Tensor 支援

DML_FEATURE_LEVEL_3_0和更新版本

張	種類	支援的維度計數	支援的資料類型
InputTensor	輸入	1 到 8	FLOAT32、FLOAT16
OutputTensor	輸出	1 到 8	FLOAT32、FLOAT16

DML_FEATURE_LEVEL_1_0和更新版本

張	種類	支援的維度計數	支援的資料類型
InputTensor	輸入	4	FLOAT32、FLOAT16
OutputTensor	輸出	4	FLOAT32、FLOAT16

規格需求

需求	值
標頭	directml.h

Share via