
LSTM工具箱已开发完成。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
LSTM工具箱是Matlab RNN库中速度最快的选择之一。其性能模型采用一个LSTM模型,该模型包含[1024, 1024, 1024]的隐藏维度,并设置了10个时间步长,同时具备256个暗淡输入的特性。所使用的硬件设备为i7-4710hq处理器和GTX940m显卡。 LSTMtoolbox在每个epoch内仅需60秒完成计算。相比之下,Keras(版本1.2.2,Tensorflow后端,cudnn5.1)在相同的条件下需要29秒/epoch完成。该产品的显著特点在于其高度的并行执行能力,它通过将四个门的权重与W矩阵对应实现高效运算,并利用批次中每个时间步的x和h值与3D张量xh进行对应匹配。此外,系统一次性计算每个样本中每个时间步的x * W,以及输入、遗忘和输出门的激活值。
全部评论 (0)
还没有任何评论哟~


