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 |