LSTM输入的维度与输出预测序列的维度相同如何实现
时间: 2023-05-15 19:04:35 浏览: 352
麻雀算法算法SSA优化LSTM实现多维输入,单维输出的预测模型,代码内注释详细,可学习性强,直接替换数据就可以用 SSA-LS
可以通过设置相同的维度大小来实现LSTM输入的维度与输出预测序列的维度相同。在LSTM中,输入序列的维度通常是[batch_size, sequence_length, input_size],而输出预测序列的维度通常是[batch_size, sequence_length, output_size]。如果要使输入和输出的维度相同,可以将output_size设置为input_size。这样,LSTM的输入和输出维度就相同了。
阅读全文