Compartir a través de

estructura DML_ELEMENT_WISE_MEAN_OPERATOR_DESC (directml.h)

  • Artículo

Promedio 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) / 2

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

Sintaxis

struct DML_ELEMENT_WISE_MEAN_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_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 4 FLOAT32, FLOAT16, INT32, UINT32
BTensor Entrada 4 FLOAT32, FLOAT16, INT32, UINT32
OutputTensor Resultados 4 FLOAT32, FLOAT16, INT32, UINT32

DML_FEATURE_LEVEL_1_0 y versiones posteriores

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

Requisitos

Requisito Valor
Header directml.h