estrutura DML_ELEMENT_WISE_ROUND_OPERATOR_DESC (directml.h)

Artigo
11/12/2024

Arredonda cada elemento de InputTensor para um valor inteiro, colocando o resultado no elemento correspondente de OutputTensor.

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_ROUND_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  DML_ROUNDING_MODE     RoundingMode;
};

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 para o qual gravar os resultados.

RoundingMode

Tipo: DML_ROUNDING_MODE

Um DML_ROUNDING_MODE determinando a direção para a qual arredondar.

Se DML_ROUNDING_MODE_HALVES_TO_NEAREST_EVEN: os valores serão arredondados para o inteiro mais próximo, com valores de metade (por exemplo, 0,5) sendo arredondados em direção ao inteiro mais próximo.
Se DML_ROUNDING_MODE_TOWARD_ZERO: os valores serão arredondados para zero. Isso trunca efetivamente a parte fracionária.
Se DML_ROUNDING_MODE_TOWARD_INFINITY: os valores serão arredondados para o inteiro mais próximo, com valores de metade (por exemplo, 0,5) sendo arredondados para longe de zero (em direção ao infinito positivo ou negativo, dependendo do sinal do valor).

Disponibilidade

Este operador foi introduzido no DML_FEATURE_LEVEL_2_1.

Restrições do Tensor

InputTensor e OutputTensor devem ter o mesmo DataType, DimensionCounte Sizes.

Suporte ao Tensor

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
OutputTensor	Saída	1 a 8	FLOAT32, FLOAT16

DML_FEATURE_LEVEL_2_1 e superior

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

Requisitos

Requisito	Valor
de cliente com suporte mínimo	Windows 10 Build 20348
servidor com suporte mínimo	Windows 10 Build 20348
cabeçalho	directml.h

Compartilhar via