DML_ELEMENT_WISE_IDENTITY_OPERATOR_DESC struttura (directml.h)

Articolo
08/23/2023

Calcola l'identità per ogni elemento di InputTensor, inserendo il risultato nell'elemento corrispondente di OutputTensor.

f(x) = x

Questo operatore supporta l'esecuzione sul posto, ovvero outputTensor è consentito per alias InputTensor durante l'associazione.

Sintassi

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

Members

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.

ScaleBias

Tipo: _Maybenull_ const DML_SCALE_BIAS*

Scala e pregiudizi facoltativo da applicare all'input. Se presente, questo ha l'effetto di applicare la funzione g(x) = x * scale + bias a ogni elemento di input prima di calcolare questo operatore.

Commenti

L'operazione di identità viene spesso usata per copiare un tensore.

Può essere usato anche per trasformare il layout dei tensori manipolando gli stride (vedere Uso di strides per esprimere il layout di spaziatura e memoria).

Disponibilità

Questo operatore è stato introdotto in DML_FEATURE_LEVEL_1_0.

Vincoli tensor

InputTensor e OutputTensor devono avere lo stesso oggetto DataType, DimensionCount e Sizes.

Supporto di Tensor

DML_FEATURE_LEVEL_4_1 e versioni successive

Tensore	Tipo	Conteggi delle dimensioni supportate	Tipi di dati supportati
InputTensor	Input	da 1 a 8	FLOAT64, FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8
OutputTensor	Output	da 1 a 8	FLOAT64, FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_3_0 e versioni successive

Tensore	Tipo	Conteggi delle dimensioni supportate	Tipi di dati supportati
InputTensor	Input	da 1 a 8	FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8
OutputTensor	Output	da 1 a 8	FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_2_1 e versioni successive

Tensore	Tipo	Conteggi delle dimensioni supportate	Tipi di dati supportati
InputTensor	Input	da 4 a 5	FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8
OutputTensor	Output	da 4 a 5	FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_1_0 e versioni successive

Tensore	Tipo	Conteggi delle dimensioni supportate	Tipi di dati supportati
InputTensor	Input	da 4 a 5	FLOAT32, FLOAT16
OutputTensor	Output	da 4 a 5	FLOAT32, FLOAT16

Requisiti

Requisito	Valore
Intestazione	directml.h

Condividi tramite