打赏
2019年7月17日
[pytorch] 自定义激活函数中的注意事项
摘要： 如何在pytorch中使用自定义的激活函数？ 如果自定义的激活函数是可导的，那么可以直接写一个python function来定义并调用，因为pytorch的autograd会自动对其求导。 如果自定义的激活函数不是可导的，比如类似于ReLU的分段可导的函数，需要写一个继承torch.autogra 阅读全文
posted @ 2019-07-17 15:53 listenviolet 阅读(8029) 评论(0) 推荐(0) 编辑