页面加载中…
AI033. LSTM 单元前向传播
困难通过率 0% · 提交 0 · 通过 0
循环网络LSTM
本题改编自算法机考高频题型 · 考点：循环网络 / LSTM · 难度：困难
实现单层长短期记忆网络（LSTM）的前向传播（输入维 = 隐藏维 = d）：输入门 i=σ(...)、遗忘门 f=σ(...)、候选值 g=tanh(...)、输出门 o=σ(...)；细胞状态 c_t = f⊙c_{t−1} + i⊙g；隐藏态 h_t = o⊙tanh(c_t)。σ 为 sigmoid 函数，σ(z) = 1/(1+e^{−z})。h_0、c_0 由输入给出，输出最终的 h_T 与 c_T。
这类题属于算法机考高频题型中「循环网络 / LSTM」方向的高频题型，通常考察对「循环网络 / LSTM」的建模能力与边界条件处理。掌握本题的解题思路后，可举一反三应对同类真题方向，稳步提升机考通过率。
输入描述
第一行两个整数 T d。随后按 i、f、g、o 的固定顺序给出四个门的参数，每个门依次为 d×d 的 W_x、d×d 的 W_h、1 行 d 个数的 b；随后 1 行 h_0、1 行 c_0；随后 T 行、每行 d 个实数为 x_1…x_T。
输出描述
第一行 d 个数为 h_T，第二行 d 个数为 c_T（四舍五入保留两位小数，单个空格分隔）；-0.00 输出 0.00。舍入规则：用语言的标准两位小数格式化（f"{x:.2f}"、printf("%.2f")、String.format("%.2f")）即可；测试数据保证每个待输出值的精确值到 0.005 舍入边界的距离不小于 1e-8，三种语言结果一致，无需自行实现舍入。
示例
示例 1
输入示例
1 1 0.00 0.00 10.00 0.00 0.00 10.00 0.00 0.00 2.00 0.00 0.00 10.00 0.00 1.00 1.00
输出示例
0.96 1.96
说明：
d=1，各门激活值接近 1
时间限制 4000 ms · 内存限制 256 MB
本平台为独立第三方培训机构，与华为技术有限公司无任何关联；课程的服务内容与权益以购买协议为准，学习效果因个人情况而异。「华为 OD」「华为可信」等仅为对岗位与考试方向的客观描述，相关商标归各自权利人所有。
本题暂无题解精讲。
我的提交
登录后可查看你在本题的历史提交，以及每次的各用例通过情况。
