小贝子编程

LSTM网络单元/节点数



我有一个神经网络，其中一些层是LSTM节点/单元。

如何计算给定模型中有多少LSTM节点/单元?

试试这个:

print(len([l for l in model.layers if isinstance(l, tf.keras.layers.LSTM)]))

相关内容

最新更新


热门标签：

javascript python java c# php android html jquery c++ css ios sql mysql arrays asp.net json python-3.x ruby-on-rails .net sql-server django objective-c excel regex ruby linux ajax iphone xml vba spring asp.net-mvc database wordpress string postgresql wpf windows xcode bash git oracle list vb.net multithreading eclipse algorithm macos powershell visual-studio image forms numpy scala function api selenium