共用方式為

DML_CAST_OPERATOR_DESC 結構 (directml.h)

  • 發行項

將輸入中的每個項目轉換成輸出張量的數據型別，並將結果儲存在輸出的對應元素中。

語法

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

成員

InputTensor

類型： const DML_TENSOR_DESC*

要寫入結果的張量。 這個 Tensor 的大小 應該符合 InputTensor。

OutputTensor

類型： const DML_TENSOR_DESC*

常 數的指標DML_TENSOR_DESC 包含要寫入結果之張量的描述。

備註

某些硬體可能不支援某些數據類型。 若要判斷是否支持數據類型，請使用 IDMLDevice：：CheckFeatureSupport 搭配 DML_FEATURE_TENSOR_DATA_TYPE_SUPPORT。

可用性

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

Tensor 條件約束

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

Tensor 支援

DML_FEATURE_LEVEL_4_1和更新版本

張 種類 支援的維度計數 支援的資料類型
InputTensor 輸入 1 到 8 FLOAT64、FLOAT32、FLOAT16、INT64、INT32、INT16、INT8、UINT64、UINT32、UINT16、UINT8
OutputTensor 輸出 1 到 8 FLOAT64、FLOAT32、FLOAT16、INT64、INT32、INT16、INT8、UINT64、UINT32、UINT16、UINT8

DML_FEATURE_LEVEL_3_1和更新版本

張 種類 支援的維度計數 支援的資料類型
InputTensor 輸入 1 到 8 FLOAT32、FLOAT16、INT64、INT32、INT16、INT8、UINT64、UINT32、UINT16、UINT8
OutputTensor 輸出 1 到 8 FLOAT32、FLOAT16、INT64、INT32、INT16、INT8、UINT64、UINT32、UINT16、UINT8

DML_FEATURE_LEVEL_3_0和更新版本

張 種類 支援的維度計數 支援的資料類型
InputTensor 輸入 4 FLOAT32、FLOAT16、INT64、INT32、INT16、INT8、UINT64、UINT32、UINT16、UINT8
OutputTensor 輸出 4 FLOAT32、FLOAT16、INT64、INT32、INT16、INT8、UINT64、UINT32、UINT16、UINT8

DML_FEATURE_LEVEL_1_0和更新版本

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

規格需求

需求 值
標頭 directml.h