上传者: 42099815
|
上传时间: 2021-10-13 11:54:14
|
文件大小: 18KB
|
文件类型: -
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类型