LSTM工具箱 源码

LSTM工具箱 最快的Matlab RNN库之一。 性能 模型:一个LSTM模型具有[1024,1024,1024]隐藏尺寸和10个时间步长,并具有256个暗淡输入。 设备:i7-4710hq,GTX940m LSTMtoolbox:60sec / epoch Keras(1.2.2,Tensorflow backend,cudnn5.1):29sec / epoch 产品特点 高并行执行。 将4个门的权重与W对应,并将批次中每个时间步的x和h值与3D张量xh对应。一次计算每个样本中每个样本的每个时间步的x * W。 一次计算输入,忘记,输出门的激活值。 OOP风格 使用struct类型