Разница между функцией GRUCell в потоке и тензорном потоке

Question

Я хочу перенести мой код из pytorch в tenorflow2.0, но я не знаю, как GRU и GRUCell в pytorch реализованы в tenorflow2.0.

self.dec_gru_a = nn.GRUCell(self.embed_size, self.hidden_size)
hidden = self.dec_gru_a(h, hidden)