Compartir a través de

estructura DML_ELEMENT_WISE_MULTIPLY_OPERATOR_DESC (directml.h)

  • Artículo

Calcula el producto de cada par de elementos correspondientes de los tensores de entrada, colocando el resultado en el elemento correspondiente de OutputTensor.

f(a, b) = a * b

Este operador admite la ejecución en contexto, lo que significa que OutputTensor tiene permiso para alias uno de los tensores de entrada durante el enlace.

Sintaxis

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

Miembros

ATensor

Tipo: const DML_TENSOR_DESC*

Tensor que contiene las entradas del lado izquierdo.

BTensor

Tipo: const DML_TENSOR_DESC*

Tensor que contiene las entradas del lado derecho.

OutputTensor

Tipo: const DML_TENSOR_DESC*

Tensor de salida en el que se van a escribir los resultados.

Disponibilidad

Este operador se introdujo en DML_FEATURE_LEVEL_1_0.

Restricciones tensor

ATensor, BTensor y OutputTensor deben tener el mismo DataType, DimensionCount y Sizes.

Compatibilidad con Tensor

DML_FEATURE_LEVEL_5_1 y versiones posteriores

Tensor Clase Recuentos de dimensiones admitidos Tipos de datos admitidos
ATensor Entrada De 1 a 8 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8
BTensor Entrada De 1 a 8 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8
OutputTensor Resultados De 1 a 8 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_4_1 y versiones posteriores

Tensor Clase Recuentos de dimensiones admitidos Tipos de datos admitidos
ATensor Entrada De 1 a 8 FLOAT32, FLOAT16, INT64, INT32, UINT64, UINT32
BTensor Entrada De 1 a 8 FLOAT32, FLOAT16, INT64, INT32, UINT64, UINT32
OutputTensor Resultados De 1 a 8 FLOAT32, FLOAT16, INT64, INT32, UINT64, UINT32

DML_FEATURE_LEVEL_3_0 y versiones posteriores

Tensor Clase Recuentos de dimensiones admitidos Tipos de datos admitidos
ATensor Entrada De 1 a 8 FLOAT32, FLOAT16, INT32, UINT32
BTensor Entrada De 1 a 8 FLOAT32, FLOAT16, INT32, UINT32
OutputTensor Resultados De 1 a 8 FLOAT32, FLOAT16, INT32, UINT32

DML_FEATURE_LEVEL_2_1 y versiones posteriores

Tensor Clase Recuentos de dimensiones admitidos Tipos de datos admitidos
ATensor Entrada De 4 a 5 FLOAT32, FLOAT16, INT32, UINT32
BTensor Entrada De 4 a 5 FLOAT32, FLOAT16, INT32, UINT32
OutputTensor Resultados De 4 a 5 FLOAT32, FLOAT16, INT32, UINT32

DML_FEATURE_LEVEL_1_0 y versiones posteriores

Tensor Clase Recuentos de dimensiones admitidos Tipos de datos admitidos
ATensor Entrada De 4 a 5 FLOAT32, FLOAT16
BTensor Entrada De 4 a 5 FLOAT32, FLOAT16
OutputTensor Resultados De 4 a 5 FLOAT32, FLOAT16

Requisitos

Requisito Valor
Header directml.h