Freigeben über

DML_CAST_OPERATOR_DESC-Struktur (directml.h)

  • Artikel

Wandelt jedes Element in der Eingabe in den Datentyp des Ausgabe tensors um und speichert das Ergebnis im entsprechenden Element der Ausgabe.

Syntax

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

Member

InputTensor

Typ: const DML_TENSOR_DESC*

Der Tensor, in den die Ergebnisse geschrieben werden sollen. Die Größen dieses Tensors sollten mit InputTensor übereinstimmen.

OutputTensor

Typ: const DML_TENSOR_DESC*

Ein Zeiger auf eine konstante DML_TENSOR_DESC , die die Beschreibung des Tensors enthält, in den die Ergebnisse geschrieben werden sollen.

Hinweise

Einige Datentypen werden auf bestimmter Hardware möglicherweise nicht unterstützt. Um zu bestimmen, ob ein Datentyp unterstützt wird, verwenden Sie IDMLDevice::CheckFeatureSupport mit DML_FEATURE_TENSOR_DATA_TYPE_SUPPORT.

Verfügbarkeit

Dieser Operator wurde in DML_FEATURE_LEVEL_1_0eingeführt.

Tensoreinschränkungen

InputTensor und OutputTensor müssen über die gleichen DimensionCount und Sizes verfügen.

Tensorunterstützung

DML_FEATURE_LEVEL_4_1 und höher

Tensor Variante Unterstützte Dimensionsanzahl Unterstützte Datentypen
InputTensor Eingabe 1 bis 8 FLOAT64, FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8
OutputTensor Ausgabe 1 bis 8 FLOAT64, FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_3_1 und höher

Tensor Variante Unterstützte Dimensionsanzahl Unterstützte Datentypen
InputTensor Eingabe 1 bis 8 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8
OutputTensor Ausgabe 1 bis 8 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_3_0 und höher

Tensor Variante Unterstützte Dimensionsanzahl Unterstützte Datentypen
InputTensor Eingabe 4 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8
OutputTensor Ausgabe 4 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_1_0 und höher

Tensor Variante Unterstützte Dimensionsanzahl Unterstützte Datentypen
InputTensor Eingabe 4 FLOAT32, FLOAT16, INT16, INT8, UINT16, UINT8
OutputTensor Ausgabe 4 FLOAT32, FLOAT16, INT16, INT8, UINT16, UINT8

Anforderungen

Anforderung Wert
Header directml.h