标签: pytorch
现在我们知道,GRU中有tanh激活。我想使用其他激活,如S型或线性激活,我应该如何在pytorch中修改nn.GRU源代码
我还没有定义新的层,我也不知道该怎么做
self.rnn = BidirectionalGRU(n_in=256, n_hidden=128)
我阅读了官方文档,没有要修改激活的内容。
我希望我可以指定激活参数