学习求助贴，现在需要模拟gru自定义一个新门控单元

xlinry

学习求助贴，现在需要模拟gru自定义一个新门控单元，输入和输出的格式及类型都参照pytorch里面的gru
传入格式为PackedSequence->{data, batch_sizes, sorted_indices, unsorted_indices }
输出格式为Tuple[PackedSequence, Tensor]
搞好久了，看pytorch源码，一层层套娃后就不是python写的了，也看不到更底层的代码
有没有大佬帮忙看看呀

或者直接手写GRU，我自己改也行，主要看不到底层代码，不知道怎么实现的。
rnn.py中GRU代码如下（截取）：
不知道如何实现的地方：
1、self._update_flat_weights()2、self.check_forward_args(input, hx, batch_sizes)3、if batch_sizes is None:
    result = _VF.gru(input, hx, self._flat_weights, self.bias, self.num_layers,
                     self.dropout, self.training, self.bidirectional, self.batch_first)
else:
    result = _VF.gru(input, batch_sizes, hx, self._flat_weights, self.bias,
                     self.num_layers, self.dropout, self.training, self.bidirectional)

ps：吾爱币可以再加