[pytorch] 自定义激活函数中的注意事项

如何在pytorch中使用自定义的激活函数？ 如果自定义的激活函数是可导的，那么可以直接写一个python function来定义并调用，因为pytorch的autograd会自动对其求导。 如果 ...