如何在pytorch中定义一个新层以及如何修改GRU的激活

时间:2019-07-26 02:23:49

标签: pytorch

现在我们知道,GRU中有tanh激活。我想使用其他激活,如S型或线性激活,我应该如何在pytorch中修改nn.GRU源代码

我还没有定义新的层,我也不知道该怎么做

 self.rnn = BidirectionalGRU(n_in=256, n_hidden=128)

我阅读了官方文档,没有要修改激活的内容。

我希望我可以指定激活参数

0 个答案:

没有答案