Pytorch 如何在顺序网络中获取激活函数的学习参数值

我正在用可学习的参数实现我的自定义激活函数。例如，这可能类似于PReLu，它是一个可学习的参数

a

培训后如何访问/查看a参数值

解决方案该问题与它是一个顺序网络这一事实有关，这有助于解决该问题

与大多数其他层一样，该层是一个，您可以使用

权重

属性直接访问权重

>>> act = nn.PReLU()
>>> act.weight
Parameter containing:
tensor([0.2500], requires_grad=True)

---

你能更具体地说明是什么阻碍了你吗？您的learnable参数是自定义激活函数类的一个属性，因此您可以像pythonthanks中的任何其他属性一样访问它，这个问题与它是一个顺序网络这一事实有关，这有助于解决这个问题