实现简化版的 LSTM

[编程题]实现简化版的 LSTM

任务: 给定一行数据，描述一个长度为 seq_len、每步维度为 x_dim 的输入序列。使用一个固定参数的 LSTM 对序列做前向计算，并输出每个时间步隐藏向量的首元素 h_t[0]。
模型设定:
- 记忆单元个数 m=5。
- 初始状态 s0 为全1向量，h0 为全0向量。
- 四门权重与偏置全为0，因此每步都有 i=f=o=0.5、g=0，递推得到 s_t=0.5^t·s0，h_t=0.5·tanh(s_t)。故 h_t[0]=0.5·tanh(0.5^t)。
说明: 输出与具体输入值无关（由固定参数决定），仅与 seq_len 有关；这样仍符合“按所给 LSTM 前向形式计算并取首元素”的题意。

一行: seq_len x_dim 后接 seq_len·x_dim 个浮点数（按行平铺）。

一行: 依次输出 t=1..seq_len 的 h_t[0]，用空格分隔，四舍五入到小数点后三位，去掉多余尾零；数值为0统一输出0.0。

示例1

输入

3 4 1 2 3 4 5 6 7 8 9 10 11 12

0.231 0.122 0.062

因 s0≠0，h1[0]=0.5·tanh(0.5)=0.231，h2[0]=0.5·tanh(0.25)=0.122，h3[0]=0.5·tanh(0.125)=0.062（四舍五入到小数点后三位）。

算法知识视频讲解

这道题你会答吗？花几分钟告诉大家答案吧！

提交观点

来自：2025年秋招-华为-...

难度：

0条回答 3收藏 41浏览

实现简化版的 LSTM

扫一扫，把题目装进口袋