Compartilhar via

estrutura DML_ELEMENT_WISE_MEAN_OPERATOR_DESC (directml.h)

Faz a média de cada par de elementos correspondentes dos tensores de entrada, colocando o resultado no elemento correspondente de OutputTensor.

f(a, b) = (a + b) / 2

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

Sintaxe

struct DML_ELEMENT_WISE_MEAN_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 no qual gravar os resultados.

Disponibilidade

Esse operador foi introduzido no DML_FEATURE_LEVEL_1_0.

Restrições do Tensor

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

Suporte ao Tensor

DML_FEATURE_LEVEL_3_0 e superior

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

DML_FEATURE_LEVEL_2_1 e superior

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

DML_FEATURE_LEVEL_1_0 e superior

Tensor Tipo Contagens de dimensões com suporte Tipos de dados com suporte
Atensor Entrada 4 FLOAT32, FLOAT16
BTensor Entrada 4 FLOAT32, FLOAT16
OutputTensor Saída 4 FLOAT32, FLOAT16

Requisitos

Requisito Valor
Cabeçalho directml.h
  • Last updated on