estrutura DML_ELEMENT_WISE_IDENTITY_OPERATOR_DESC (directml.h)

  • Artigo

Calcula a identidade de cada elemento de InputTensor, colocando o resultado no elemento correspondente de OutputTensor.

f(x) = x

Esse operador dá suporte à execução in-loco, o que significa que OutputTensor tem permissão para alias InputTensor durante a associação.

Sintaxe

struct DML_ELEMENT_WISE_IDENTITY_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  const DML_SCALE_BIAS  *ScaleBias;
};

Membros

InputTensor

Tipo: const DML_TENSOR_DESC*

O tensor de entrada do qual ler.

OutputTensor

Tipo: const DML_TENSOR_DESC*

O tensor de saída no qual gravar os resultados.

ScaleBias

Tipo: _Maybenull_ const DML_SCALE_BIAS*

Uma escala e um desvio opcionais a serem aplicados à entrada. Se presente, isso terá o efeito de aplicar a função g(x) = x * scale + bias a cada elemento de entrada antes de calcular esse operador.

Comentários

A operação de identidade geralmente é usada para copiar um tensor.

Ele também pode ser usado para transformar o layout de tensores manipulando passos (consulte Usando passos para expressar o preenchimento e o layout de memória).

Disponibilidade

Esse operador foi introduzido no DML_FEATURE_LEVEL_1_0.

Restrições do Tensor

InputTensor e OutputTensor devem ter os mesmos DataType, 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_0 e superior

Tensor Tipo Contagens de dimensões com suporte Tipos de dados com suporte
InputTensor Entrada 1 a 8 FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8
OutputTensor Saída 1 a 8 FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_2_1 e superior

Tensor Tipo Contagens de dimensões com suporte Tipos de dados com suporte
InputTensor Entrada 4 a 5 FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8
OutputTensor Saída 4 a 5 FLOAT32, FLOAT16, INT32, INT16, INT8, 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 a 5 FLOAT32, FLOAT16
OutputTensor Saída 4 a 5 FLOAT32, FLOAT16

Requisitos

Requisito Valor
Cabeçalho directml.h