如何在pytorch中修改rnn单元格？

作者: 你抱着孩子@先跑
发布时间: 2024-12-25 03:08:13 (13天前)
转自：

2 条回复

0#
回复此人
早岁那知世事艰 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 是的，你实现它“通过for或while循环”。由于Pytorch 1.0有JIT <a href="https://pytorch.org/docs/stable/jit.html" rel="nofollow noreferrer"> https://pytorch.org/docs/stable/jit.html </A> 这很好用（最好使用PyTorch的最新git版本，因为最近改进了JIT），并且取决于你的网络和实现，可以像原生的PyTorch C ++实现一样快（但仍然比CuDNN慢）。 </p> <P> 你可以看到示例实现 <a href="https://github.com/pytorch/benchmark/blob/master/rnns/fastrnns/custom_lstms.py" rel="nofollow noreferrer"> https://github.com/pytorch/benchmark/blob/master/rnns/fastrnns/custom_lstms.py </A> </p> </DIV>

编辑

登录后才能参与评论