0
点赞
收藏
分享

微信扫一扫

【PyTorch基础】PReLU

笑望叔叔 2022-07-13 阅读 55
编程语言

api:

CLASS ​​torch.nn.​​​​PReLU​​(num_parameters=1init=0.25device=Nonedtype=None)

注意参数和ReLU不一样的;

原理:

PReLU(x)=max(0,x)+a∗min(0,x)

其中,a是一个学习参数,默认a是对所有通道的,如果参数有通道数,则不同的a是对应不同通道的;

 

参考

1. ​​PReLU_pytorch​​;

2. 

举报

相关推荐

0 条评论