LSTM神经网络输入输出究竟是怎样的?
创始人
2024-11-08 04:18:30
LSTM神经网络输入输出究竟是怎样的?

常规feedforward输入和输出:矩阵输入矩阵形状:(n_samples,dim_input)输出矩阵形状:(n_samples,dim_output)注:真正测试/训练的时候,网络的输入和输出就是向量而已。加入n_samples这个维度是为了可以实现一次训练多个样本,求出平均梯度来更新权重,这个叫做Mini-batchgradientdescent。如果n_samples等于1,那么这种更新方式叫做StochasticGradientDescent(SGD)。Feedforward的输入输出的本质都是单个向量。常规Recurrent(RNN/LSTM/GRU)输入和输出:张量输入张量形状:(time_steps,n_samples,dim_input)输出张量形状:(time_steps,n_samples,dim_output)注:同样是保留了Mini-batchgradientdescent的训练方式,但不同之处在于多了timestep这个维度。Recurrent的任意时刻的输入的本质还是单个向量,只不过是将不同时刻的向量按顺序输入网络。所以你可能更愿意理解为一串向量asequenceofvectors,或者是矩阵。

相关内容

热门资讯

芯朋微发布12款面向AI服务器... 12月5日,国内领先的功率系统芯片设计企业芯朋微电子(芯朋微,688508)在深圳发布12款面向AI...
杨振宁、霍金、居里夫人在物理学... 我用8种AI给物理学家打分,之后取总分,最终获得了一个相对客观的物理学前50名排行榜。 打分之前先要...
C罗,投了家AI创业公司 文:韦亚军 摄影:Bob C罗的可投资资产高达8亿美元。 近日,葡萄牙球星C罗本人(克里斯蒂亚诺·...