Compartir a través de


estructura DML_ELEMENT_WISE_ROUND_OPERATOR_DESC (directml.h)

Redondea cada elemento de InputTensor a un valor entero, colocando el resultado en el elemento correspondiente de OutputTensor.

Este operador admite la ejecución local, lo que significa que OutputTensor se permite alias inputTensor durante el enlace.

Sintaxis

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

Miembros

InputTensor

Tipo: const DML_TENSOR_DESC*

Tensor de entrada desde el que se va a leer.

OutputTensor

Tipo: const DML_TENSOR_DESC*

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

RoundingMode

Tipo: DML_ROUNDING_MODE

Un DML_ROUNDING_MODE determinar la dirección hacia la que redondear.

  • Si DML_ROUNDING_MODE_HALVES_TO_NEAREST_EVEN: los valores se redondean al entero más cercano, con valores de mitad de camino (por ejemplo, 0,5) redondeados hacia el entero par más cercano.
  • Si DML_ROUNDING_MODE_TOWARD_ZERO: los valores se redondean hacia cero. Esto trunca eficazmente la parte fraccionarcional.
  • Si DML_ROUNDING_MODE_TOWARD_INFINITY: los valores se redondean al entero más cercano, con valores de mitad de camino (por ejemplo, 0,5) redondeados lejos de cero (hacia infinito positivo o negativo, dependiendo del signo del valor).

Disponibilidad

Este operador se introdujo en DML_FEATURE_LEVEL_2_1.

Restricciones tensor

InputTensor y OutputTensor deben tener el mismo DataType, DimensionCounty Sizes.

Compatibilidad con Tensor

DML_FEATURE_LEVEL_3_0 y versiones posteriores

Tensor Amable Recuentos de dimensiones admitidos Tipos de datos admitidos
InputTensor Entrada De 1 a 8 FLOAT32, FLOAT16
OutputTensor Salida De 1 a 8 FLOAT32, FLOAT16

DML_FEATURE_LEVEL_2_1 y versiones posteriores

Tensor Amable Recuentos de dimensiones admitidos Tipos de datos admitidos
InputTensor Entrada 4 FLOAT32, FLOAT16
OutputTensor Salida 4 FLOAT32, FLOAT16

Requisitos

Requisito Valor
cliente mínimo admitido Compilación 20348 de Windows 10
servidor mínimo admitido Compilación 20348 de Windows 10
encabezado de directml.h