首页 > 解决方案 > pytorch和tensorflow中GRUCell函数的区别

问题描述

我想把我的代码从pytorch迁移到tensorflow2.0,但是不知道pytorch中的GRU和GRUCell是如何在tensorflow2.0中实现的。

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

标签: pythonpytorch

解决方案


推荐阅读