关于pytorch中forward函数的写法?

比如使用pytorch写个rnn模型,看过很多例子,forward的参数都是x,h_state两个,但是看过一个代码(github.com/zhangxu0307/time_series_forecasting_pytorch/blob/master/code/model.py),不是这么写的,我可能对于rnn理解不够透彻,这样的写法比较新颖,谁能给解释下这样对不对?
已邀请:

16块的黄鹤楼

赞同来自:

执行operation。

所有的Function子类都需要重写这个方法。
可以接收和返回任意个数 tensors
 
我觉得你可以看看pytorch文档

要回复问题请先登录注册