DML_ACTIVATION_HARDMAX_OPERATOR_DESC結構 (directml.h)

在 InputTensor 的每個元素上執行 hardmax 函式，並將結果放入 OutputTensor 的對應元素。

運算符會針對圖層中第一次出現的最大值計算 hardmax (1，而針對指定輸入中每個數據列的所有其他值) 0。

語法

struct DML_ACTIVATION_HARDMAX_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
};

成員

InputTensor

類型： const DML_TENSOR_DESC*

要從中讀取輸入的張量。 此張量必須具有不大於 2 的有效順位 。 張量的有效排名是 tensor 的 DimensionCount ，不包括大小為 1 的最左邊維度。 例如，的 { 1, 1, BatchCount, Width } 張量大小有效，且相當於 大小的 { BatchCount, Width }張量。

OutputTensor

類型： const DML_TENSOR_DESC*

要寫入結果的輸出張量。

備註

運算符會計算第一個最大值的 hardmax (1，而所有其他) 層在指定輸入批次中每個圖層的值則為 0。 輸入是 2D 張量 (Tensor) 大小 (batch_size x input_feature_dimensions) 。 輸出張量具有相同的圖形，且包含對應輸入的 hardmax 值。

可用性

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

Tensor 條件約束

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

Tensor 支援

DML_FEATURE_LEVEL_3_0和更新版本

DML_FEATURE_LEVEL_2_0和更新版本

DML_FEATURE_LEVEL_1_0和更新版本

規格需求

另請參閱

DML_ARGMAX_OPERATOR_DESC結構