estructura DML_ACTIVATION_THRESHOLDED_RELU_OPERATOR_DESC (directml.h)
Realiza una función de activación de unidad lineal rectificada (ReLU) umbral en cada elemento de InputTensor, colocando el resultado en el elemento correspondiente de OutputTensor.
f(x) = x, if x > Alpha
0, otherwise
Este operador admite la ejecución local, lo que significa que el tensor de salida puede alias InputTensor durante el enlace.
Sintaxis
struct DML_ACTIVATION_THRESHOLDED_RELU_OPERATOR_DESC {
const DML_TENSOR_DESC *InputTensor;
const DML_TENSOR_DESC *OutputTensor;
FLOAT Alpha;
};
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.
Alpha
Tipo: FLOAT
Umbral de la función. Un valor predeterminado típico para este valor es 1.0.
Disponibilidad
Este operador se introdujo en DML_FEATURE_LEVEL_1_0.
Restricciones tensor
InputTensor y OutputTensor deben tener el mismo DataType, DimensionCount y Sizes.
Compatibilidad con Tensor
DML_FEATURE_LEVEL_3_0 y versiones posteriores
| Tensor | Kind | Recuentos de dimensiones admitidos | Tipos de datos admitidos |
|---|---|---|---|
| InputTensor | Entrada | De 1 a 8 | FLOAT32, FLOAT16 |
| OutputTensor | Resultados | De 1 a 8 | FLOAT32, FLOAT16 |
DML_FEATURE_LEVEL_2_0 y versiones posteriores
| Tensor | Kind | Recuentos de dimensiones admitidos | Tipos de datos admitidos |
|---|---|---|---|
| InputTensor | Entrada | De 4 a 5 | FLOAT32, FLOAT16 |
| OutputTensor | Resultados | De 4 a 5 | FLOAT32, FLOAT16 |
DML_FEATURE_LEVEL_1_0 y versiones posteriores
| Tensor | Kind | Recuentos de dimensiones admitidos | Tipos de datos admitidos |
|---|---|---|---|
| InputTensor | Entrada | 4 | FLOAT32, FLOAT16 |
| OutputTensor | Resultados | 4 | FLOAT32, FLOAT16 |
Requisitos
| Encabezado | directml.h |