Compartilhar via

estrutura DML_ELEMENT_WISE_DIFFERENCE_SQUARE_OPERATOR_DESC (directml.h)

  • Artigo

Subtrai cada elemento de BTensor do elemento correspondente de ATensor, multiplica o resultado por si só e coloca o resultado no elemento correspondente de OutputTensor.

f(a, b) = (a - b) * (a - b)

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

Sintaxe

struct DML_ELEMENT_WISE_DIFFERENCE_SQUARE_OPERATOR_DESC {
  const DML_TENSOR_DESC *ATensor;
  const DML_TENSOR_DESC *BTensor;
  const DML_TENSOR_DESC *OutputTensor;
};

Membros

ATensor

Tipo: const DML_TENSOR_DESC*

Um tensor que contém as entradas do lado esquerdo.

BTensor

Tipo: const DML_TENSOR_DESC*

Um tensor que contém as entradas do lado direito.

OutputTensor

Tipo: const DML_TENSOR_DESC*

O tensor de saída para o qual gravar os resultados.

Comentários

Disponibilidade

Esse operador foi introduzido em DML_FEATURE_LEVEL_3_1.

Restrições do Tensor

ATensor, BTensor e OutputTensor devem ter os mesmos DataType, DimensionCount e Sizes.

Suporte ao Tensor

Tensor Tipo Contagens de dimensões com suporte Tipos de dados com suporte
Atensor Entrada 1 a 8 FLOAT32, FLOAT16, INT32, UINT32
BTensor Entrada 1 a 8 FLOAT32, FLOAT16, INT32, UINT32
OutputTensor Saída 1 a 8 FLOAT32, FLOAT16, INT32, UINT32

Requisitos

   
Cliente mínimo com suporte Windows Build 22000
Servidor mínimo com suporte Windows Build 22000
Cabeçalho directml.h