Intel® oneAPI Deep Neural Network Developer Guide and Reference
PReLU
General
PReLU operation performs element-wise parametric ReLU operation on a given input tensor, based on the following mathematical formula:
 
   Operation attributes
Attribute Name  |  
        Description  |  
        Value Type  |  
        Supported Values  |  
        Required or Optional  |  
       
|---|---|---|---|---|
Denotes the data format of the input and output data.  |  
        string  |  
        NCX , NXC (default)  |  
        Optional  |  
       |
Denotes whether to apply per_channel broadcast when slope is 1D tensor.  |  
        bool  |  
        false , true (default)  |  
        Optional  |  
       
Broadcasting Rules
Only slope tensor supports broadcasting semantics. Slope tensor is uni-directionally broadcasted to 
 if one of the following rules is met:
1: slope is 1D tensor and per_channel_broadcast is set to true, the length of slope tensor is equal to the length of
 of channel dimension.2: slope is 1D tensor and per_channel_broadcast is set to false, the length of slope tensor is equal to the length of
 of the rightmost dimension.3: slope is nD tensor, starting from the rightmost dimension,
 or 
 or slope dimension i is empty.
Execution arguments
The inputs and outputs must be provided according to below index order when constructing an operation.
Inputs
Index  |  
        Argument Name  |  
        Required or Optional  |  
       
|---|---|---|
0  |  
        src  |  
        Required  |  
       
1  |  
        slope  |  
        Required  |  
       
Outputs
Index  |  
        Argument Name  |  
        Required or Optional  |  
       
|---|---|---|
0  |  
        dst  |  
        Required  |  
       
Supported data types
PReLU operation supports the following data type combinations.
Src  |  
        Dst  |  
        Slope  |  
       
|---|---|---|
f32  |  
        f32  |  
        f32  |  
       
bf16  |  
        bf16  |  
        bf16  |  
       
f16  |  
        f16  |  
        f16  |