昇腾社区首页
中文
注册

PRelu

功能

PRelu激活函数。Tensor中value大于0,则取该value,小于0时取权重与value的乘积。

输入

  • x:输入Tensor,数据类型:float16、float。
  • slope:输入Tensor,数据类型和输入x一致。

输出

y:输出Tensor,和输入x同样的数据类型和shape。

约束

slope必须是1维,当输入x的shape是1维时,slope的维度值必须为1;输入x的shape是其他维度时,slope的维度值可以为1或者为输入x的shape[1]。

支持的ONNX版本

Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18