struttura DML_ELEMENT_WISE_ROUND_OPERATOR_DESC (directml.h)

Articolo
11/12/2024

Arrotonda ogni elemento di InputTensor a un valore intero, inserendo il risultato nell'elemento corrispondente di OutputTensor.

Questo operatore supporta l'esecuzione sul posto, vale a dire che outputTensor è consentito alias InputTensor durante l'associazione.

Sintassi

struct DML_ELEMENT_WISE_ROUND_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  DML_ROUNDING_MODE     RoundingMode;
};

Membri

InputTensor

Tipo: const DML_TENSOR_DESC*

Tensore di input da cui leggere.

OutputTensor

Tipo: const DML_TENSOR_DESC*

Tensore di output in cui scrivere i risultati.

RoundingMode

Tipo: DML_ROUNDING_MODE

Oggetto DML_ROUNDING_MODE determinare la direzione verso cui arrotondarsi.

Se DML_ROUNDING_MODE_HALVES_TO_NEAREST_EVEN: i valori vengono arrotondati all'intero più vicino, con valori a metà (ad esempio, 0,5) arrotondati al numero intero pari più vicino.
Se DML_ROUNDING_MODE_TOWARD_ZERO: i valori vengono arrotondati verso zero. In questo modo la parte frazionaria viene troncata in modo efficace.
Se DML_ROUNDING_MODE_TOWARD_INFINITY: i valori vengono arrotondati all'intero più vicino, con valori a metà (ad esempio, 0,5) arrotondati a distanza da zero (verso infinito positivo o negativo, a seconda del segno del valore).

Disponibilità

Questo operatore è stato introdotto in DML_FEATURE_LEVEL_2_1.

Vincoli tensor

InputTensor e OutputTensor devono avere lo stesso DataType, DimensionCounte dimensioni .

Supporto tensor

DML_FEATURE_LEVEL_3_0 e versioni successive

Tensore	Gentile	Conteggi delle dimensioni supportati	Tipi di dati supportati
InputTensor	Immissione	Da 1 a 8	FLOAT32, FLOAT16
OutputTensor	Prodotto	Da 1 a 8	FLOAT32, FLOAT16

DML_FEATURE_LEVEL_2_1 e versioni successive

Tensore	Gentile	Conteggi delle dimensioni supportati	Tipi di dati supportati
InputTensor	Immissione	4	FLOAT32, FLOAT16
OutputTensor	Prodotto	4	FLOAT32, FLOAT16

Fabbisogno

Requisito	Valore
client minimo supportato	Windows 10 Build 20348
server minimo supportato	Windows 10 Build 20348
intestazione	directml.h

Condividi tramite