Partager via

structure DML_CAST_OPERATOR_DESC (directml.h)

  • Article

Caste chaque élément de l’entrée en type de données du tenseur de sortie et stocke le résultat dans l’élément correspondant de la sortie.

Syntaxe

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

Membres

InputTensor

Type : const DML_TENSOR_DESC*

Tenseur dans lequel écrire les résultats. Les tailles de ce tenseur doivent correspondre à InputTensor.

OutputTensor

Type : const DML_TENSOR_DESC*

Pointeur vers une constante DML_TENSOR_DESC contenant la description du tenseur dans lequel écrire les résultats.

Remarques

Certains types de données peuvent ne pas être pris en charge sur certains matériels. Pour déterminer si un type de données est pris en charge, utilisez IDMLDevice ::CheckFeatureSupport avec DML_FEATURE_TENSOR_DATA_TYPE_SUPPORT.

Disponibilité

Cet opérateur a été introduit dans DML_FEATURE_LEVEL_1_0.

Contraintes tensoriels

InputTensor et OutputTensor doivent avoir les mêmes DimensionCount et Sizes.

Prise en charge de Tensor

DML_FEATURE_LEVEL_4_1 et versions ultérieures

Tenseur Genre Nombre de dimensions pris en charge Types de données pris en charge
InputTensor Entrée 1 à 8 FLOAT64, FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8
OutputTensor Sortie 1 à 8 FLOAT64, FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_3_1 et versions ultérieures

Tenseur Genre Nombre de dimensions pris en charge Types de données pris en charge
InputTensor Entrée 1 à 8 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8
OutputTensor Sortie 1 à 8 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_3_0 et versions ultérieures

Tenseur Genre Nombre de dimensions pris en charge Types de données pris en charge
InputTensor Entrée 4 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8
OutputTensor Sortie 4 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_1_0 et versions ultérieures

Tenseur Genre Nombre de dimensions pris en charge Types de données pris en charge
InputTensor Entrée 4 FLOAT32, FLOAT16, INT16, INT8, UINT16, UINT8
OutputTensor Sortie 4 FLOAT32, FLOAT16, INT16, INT8, UINT16, UINT8

Configuration requise

Condition requise Valeur
En-tête directml.h