estrutura DML_CAST_OPERATOR_DESC (directml.h)
Converte cada elemento na entrada para o tipo de dados do tensor de saída e armazena o resultado no elemento correspondente da saída.
Sintaxe
struct DML_CAST_OPERATOR_DESC {
const DML_TENSOR_DESC *InputTensor;
const DML_TENSOR_DESC *OutputTensor;
};
Membros
InputTensor
Tipo: const DML_TENSOR_DESC*
O tensor no qual gravar os resultados. Os Tamanhos desse tensor devem corresponder a InputTensor.
OutputTensor
Tipo: const DML_TENSOR_DESC*
Um ponteiro para uma constante DML_TENSOR_DESC que contém a descrição do tensor no qual gravar os resultados.
Comentários
Alguns tipos de dados podem não ter suporte em determinados hardwares. Para determinar se há suporte para um tipo de dados, use IDMLDevice::CheckFeatureSupport com DML_FEATURE_TENSOR_DATA_TYPE_SUPPORT.
Disponibilidade
Esse operador foi introduzido no DML_FEATURE_LEVEL_1_0.
Restrições do Tensor
InputTensor e OutputTensor devem ter os mesmos DimensionCount e Sizes.
Suporte ao Tensor
DML_FEATURE_LEVEL_4_1 e superior
| Tensor | Tipo | Contagens de dimensões com suporte | Tipos de dados com suporte |
|---|---|---|---|
| InputTensor | Entrada | 1 a 8 | FLOAT64, FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8 |
| OutputTensor | Saída | 1 a 8 | FLOAT64, FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8 |
DML_FEATURE_LEVEL_3_1 e superior
| Tensor | Tipo | Contagens de dimensões com suporte | Tipos de dados com suporte |
|---|---|---|---|
| InputTensor | Entrada | 1 a 8 | FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8 |
| OutputTensor | Saída | 1 a 8 | FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8 |
DML_FEATURE_LEVEL_3_0 e superior
| Tensor | Tipo | Contagens de dimensões com suporte | Tipos de dados com suporte |
|---|---|---|---|
| InputTensor | Entrada | 4 | FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8 |
| OutputTensor | Saída | 4 | FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8 |
DML_FEATURE_LEVEL_1_0 e superior
| Tensor | Tipo | Contagens de dimensões com suporte | Tipos de dados com suporte |
|---|---|---|---|
| InputTensor | Entrada | 4 | FLOAT32, FLOAT16, INT16, INT8, UINT16, UINT8 |
| OutputTensor | Saída | 4 | FLOAT32, FLOAT16, INT16, INT8, UINT16, UINT8 |
Requisitos
| Requisito | Valor |
|---|---|
| Cabeçalho | directml.h |