Intel® oneAPI Deep Neural Network Developer Guide and Reference
A newer version of this document is available. Customers should click here to go to the newest version.
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:
 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. 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. of the rightmost dimension.
- 3: slope is nD tensor, starting from the rightmost dimension,  or or or slope dimension i is empty. 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 |