任务: 给定一行数据,描述一个长度为 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
说明
因 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(四舍五入到小数点后三位)。
加载中...