Compartir a través de

estructura DML_CAST_OPERATOR_DESC (directml.h)

  • Artículo

Convierte cada elemento de la entrada en el tipo de datos del tensor de salida y almacena el resultado en el elemento correspondiente de la salida.

Sintaxis

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

Miembros

InputTensor

Tipo: const DML_TENSOR_DESC*

Tensor en el que se van a escribir los resultados. Los tamaños de este tensor deben coincidir con InputTensor.

OutputTensor

Tipo: const DML_TENSOR_DESC*

Puntero a una constante DML_TENSOR_DESC que contiene la descripción del tensor en el que se van a escribir los resultados.

Comentarios

Es posible que algunos tipos de datos no se admita en determinados hardware. Para determinar si se admite un tipo de datos, use IDMLDevice::CheckFeatureSupport con DML_FEATURE_TENSOR_DATA_TYPE_SUPPORT.

Disponibilidad

Este operador se introdujo en DML_FEATURE_LEVEL_1_0.

Restricciones tensor

InputTensor y OutputTensor deben tener el mismo DimensionCount y Sizes.

Compatibilidad con Tensor

DML_FEATURE_LEVEL_4_1 y versiones posteriores

Tensor Clase Recuentos de dimensiones admitidos Tipos de datos admitidos
InputTensor Entrada De 1 a 8 FLOAT64, FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8
OutputTensor Resultados De 1 a 8 FLOAT64, FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_3_1 y versiones posteriores

Tensor Clase Recuentos de dimensiones admitidos Tipos de datos admitidos
InputTensor Entrada De 1 a 8 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8
OutputTensor Resultados De 1 a 8 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_3_0 y versiones posteriores

Tensor Clase Recuentos de dimensiones admitidos Tipos de datos admitidos
InputTensor Entrada 4 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8
OutputTensor Resultados 4 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_1_0 y versiones posteriores

Tensor Clase Recuentos de dimensiones admitidos Tipos de datos admitidos
InputTensor Entrada 4 FLOAT32, FLOAT16, INT16, INT8, UINT16, UINT8
OutputTensor Resultados 4 FLOAT32, FLOAT16, INT16, INT8, UINT16, UINT8

Requisitos

Requisito Valor
Header directml.h