Condividi tramite

struttura DML_ELEMENT_WISE_MULTIPLY_OPERATOR_DESC (directml.h)

  • Articolo

Calcola il prodotto di ogni coppia di elementi corrispondenti dei tensori di input, inserendo il risultato nell'elemento corrispondente di OutputTensor.

f(a, b) = a * b

Questo operatore supporta l'esecuzione sul posto, ovvero OutputTensor è autorizzato ad eseguire l'alias di uno dei tensori di input durante l'associazione.

Sintassi

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

Members

ATensor

Tipo: const DML_TENSOR_DESC*

Tensore contenente gli input sul lato sinistro.

BTensor

Tipo: const DML_TENSOR_DESC*

Tensore contenente gli input sul lato destro.

OutputTensor

Tipo: const DML_TENSOR_DESC*

Tensore di output in cui scrivere i risultati.

Disponibilità

Questo operatore è stato introdotto in DML_FEATURE_LEVEL_1_0.

Vincoli tensor

ATensor, BTensor e OutputTensor devono avere gli stessi oggetti DataType, DimensionCount e Size.

Supporto di Tensor

DML_FEATURE_LEVEL_5_1 e versioni successive

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

DML_FEATURE_LEVEL_4_1 e versioni successive

Tensore Tipo Conteggi delle dimensioni supportati Tipi di dati supportati
ATensor Input da 1 a 8 FLOAT32, FLOAT16, INT64, INT32, UINT64, UINT32
BTensor Input da 1 a 8 FLOAT32, FLOAT16, INT64, INT32, UINT64, UINT32
OutputTensor Output da 1 a 8 FLOAT32, FLOAT16, INT64, INT32, UINT64, UINT32

DML_FEATURE_LEVEL_3_0 e versioni successive

Tensore Tipo Conteggi delle dimensioni supportati Tipi di dati supportati
ATensor Input da 1 a 8 FLOAT32, FLOAT16, INT32, UINT32
BTensor Input da 1 a 8 FLOAT32, FLOAT16, INT32, UINT32
OutputTensor Output da 1 a 8 FLOAT32, FLOAT16, INT32, UINT32

DML_FEATURE_LEVEL_2_1 e versioni successive

Tensore Tipo Conteggi delle dimensioni supportati Tipi di dati supportati
ATensor Input da 4 a 5 FLOAT32, FLOAT16, INT32, UINT32
BTensor Input da 4 a 5 FLOAT32, FLOAT16, INT32, UINT32
OutputTensor Output da 4 a 5 FLOAT32, FLOAT16, INT32, UINT32

DML_FEATURE_LEVEL_1_0 e versioni successive

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

Requisiti

Requisito Valore
Intestazione directml.h