个人中心
gpt4 book ai didi

pytorch自定义不可导激活函数的操作

转载 作者：qq735679552 更新时间：2022-09-27 22:32:09 29 4
gpt4 key购买 nike

CFSDN坚持开源创造价值，我们致力于搭建一个资源共享平台，让每一个IT人在这里找到属于你的精彩世界.

这篇CFSDN的博客文章pytorch自定义不可导激活函数的操作由作者收集整理，如果你对这篇文章有兴趣，记得点赞哟.

pytorch自定义不可导激活函数

今天自定义不可导函数的时候遇到了一个大坑.

首先我需要自定义一个函数：sign_f 。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import torch
from torch.autograd import Function
import torch.nn as nn
class sign_f(Function):
     @staticmethod
     def forward(ctx, inputs):
         output = inputs.new(inputs.size())
         output[inputs > = 0. ] = 1
         output[inputs < 0. ] = - 1
         ctx.save_for_backward(inputs)
         return output
 
     @staticmethod
     def backward(ctx, grad_output):
         input_, = ctx.saved_tensors
         grad_output[input_> 1. ] = 0
         grad_output[input_< - 1. ] = 0
         return grad_output

然后我需要把它封装为一个module 类型，就像 nn.Conv2d 模块 封装 f.conv2d 一样，于是 。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import torch
from torch.autograd import Function
import torch.nn as nn
class sign_(nn.Module):
     # 我需要的module
     def __init__( self , * kargs, * * kwargs):
         super (sign_, self ).__init__( * kargs, * * kwargs)
        
     def forward( self , inputs):
         # 使用自定义函数
         outs = sign_f(inputs)
         return outs
 
class sign_f(Function):
     @staticmethod
     def forward(ctx, inputs):
         output = inputs.new(inputs.size())
         output[inputs > = 0. ] = 1
         output[inputs < 0. ] = - 1
         ctx.save_for_backward(inputs)
         return output
 
     @staticmethod
     def backward(ctx, grad_output):
         input_, = ctx.saved_tensors
         grad_output[input_> 1. ] = 0
         grad_output[input_< - 1. ] = 0
         return grad_output

结果报错 。

TypeError: backward() missing 2 required positional arguments: 'ctx' and 'grad_output' 。

我试了半天，发现自定义函数后面要加 apply ，详细见下面 。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import torch
from torch.autograd import Function
import torch.nn as nn
class sign_(nn.Module):
 
     def __init__( self , * kargs, * * kwargs):
         super (sign_, self ).__init__( * kargs, * * kwargs)
         self .r = sign_f. apply ### <-----注意此处
        
     def forward( self , inputs):
         outs = self .r(inputs)
         return outs
 
class sign_f(Function):
     @staticmethod
     def forward(ctx, inputs):
         output = inputs.new(inputs.size())
         output[inputs > = 0. ] = 1
         output[inputs < 0. ] = - 1
         ctx.save_for_backward(inputs)
         return output
 
     @staticmethod
     def backward(ctx, grad_output):
         input_, = ctx.saved_tensors
         grad_output[input_> 1. ] = 0
         grad_output[input_< - 1. ] = 0
         return grad_output

问题解决了！ 。

PyTorch自定义带学习参数的激活函数（如sigmoid）

有的时候我们需要给损失函数设一个超参数但是又不想设固定阈值想和网络一起自动学习，例如给Sigmoid一个参数alpha进行调节 。

pytorch自定义不可导激活函数的操作

pytorch自定义不可导激活函数的操作

函数如下:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
import torch.nn as nn
import torch
class LearnableSigmoid(nn.Module):
     def __init__( self , ):
         super (LearnableSigmoid, self ).__init__()
         self .weight = torch.nn.Parameter(torch.FloatTensor( 1 ), requires_grad = True )
 
         self .reset_parameters()
     def reset_parameters( self ):
         self .weight.data.fill_( 1.0 )
        
     def forward( self , input ):
         return 1 / ( 1 +  torch.exp( - self .weight * input ))

验证和Sigmoid的一致性 。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
class LearnableSigmoid(nn.Module):
     def __init__( self , ):
         super (LearnableSigmoid, self ).__init__()
         self .weight = torch.nn.Parameter(torch.FloatTensor( 1 ), requires_grad = True )
 
         self .reset_parameters()
     def reset_parameters( self ):
         self .weight.data.fill_( 1.0 )
        
     def forward( self , input ):
         return 1 / ( 1 +  torch.exp( - self .weight * input ))
   
Sigmoid = nn.Sigmoid()
LearnSigmoid = LearnableSigmoid()
input = torch.tensor([[ 0.5289 , 0.1338 , 0.3513 ],
         [ 0.4379 , 0.1828 , 0.4629 ],
         [ 0.4302 , 0.1358 , 0.4180 ]])
 
print (Sigmoid( input ))
print (LearnSigmoid( input ))

输出结果 。

tensor([[0.6292, 0.5334, 0.5869], [0.6078, 0.5456, 0.6137], [0.6059, 0.5339, 0.6030]]) tensor([[0.6292, 0.5334, 0.5869], [0.6078, 0.5456, 0.6137], [0.6059, 0.5339, 0.6030]], grad_fn=<MulBackward0>) 。

验证权重是不是会更新 。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
import torch.nn as nn
import torch
import torch.optim as optim
class LearnableSigmoid(nn.Module):
     def __init__( self , ):
         super (LearnableSigmoid, self ).__init__()
         self .weight = torch.nn.Parameter(torch.FloatTensor( 1 ), requires_grad = True )
 
         self .reset_parameters()
 
     def reset_parameters( self ):
         self .weight.data.fill_( 1.0 )
        
     def forward( self , input ):
         return 1 / ( 1 +  torch.exp( - self .weight * input ))
        
class Net(nn.Module):
     def __init__( self ):
         super (Net, self ).__init__()      
         self .LSigmoid = LearnableSigmoid()
     def forward( self , x):               
         x = self .LSigmoid(x)
         return x
 
net = Net() 
print ( list (net.parameters()))
optimizer = optim.SGD(net.parameters(), lr = 0.01 )
learning_rate = 0.001
input_data = torch.randn( 10 , 2 )
target = torch.FloatTensor( 10 , 2 ).random_( 8 )
criterion = torch.nn.MSELoss( reduce = True , size_average = True )
 
for i in range ( 2 ):
     optimizer.zero_grad()    
     output = net(input_data)  
     loss = criterion(output, target)
     loss.backward()            
     optimizer.step()          
     print ( list (net.parameters()))

输出结果 。

tensor([1.], requires_grad=True)] [Parameter containing: tensor([0.9979], requires_grad=True)] [Parameter containing: tensor([0.9958], requires_grad=True)] 。

会更新~ 。

以上为个人经验，希望能给大家一个参考，也希望大家多多支持我.

原文链接：https://blog.csdn.net/qq_43110298/article/details/115032262 。

最后此篇关于pytorch自定义不可导激活函数的操作的文章就讲到这里了,如果你想了解更多关于pytorch自定义不可导激活函数的操作的内容请搜索CFSDN的文章或继续浏览相关文章，希望大家以后支持我的博客！ 。

29 4 0
文章推荐： IntelliJ IDEA引入第三方jar包或查看Java源码的时候报decompiled.class file bytecode version:52.0(java 8)错误的解决办法
文章推荐： MySQL中存储时间的最佳实践指南
文章推荐： IntelliJ IDEA中出现＂PSI and index do not match＂错误的解决办法
文章推荐： IntelliJ IDEA修改新建文件自动生成注释的user名
qq735679552
个人简介

我是一名优秀的程序员,十分优秀！

滴滴打车优惠券免费领取
滴滴打车优惠券
全站热门文章
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com