如何在keras的LSTM中设置隐藏状态向量的大小？

Question

我目前正在使用model.add(LSTM(50))来设置矢量大小，即在units属性中设置值，但是我非常怀疑它的正确性（在keras文档中，单位被解释为输出空间的维数）。有人可以在这里帮助我吗？

Answer 1

如果按矢量大小表示层中的节点数，那么是的，您做对了。图层的输出尺寸 与节点数相同。同样的情况适用于卷积层，滤镜数量和沿最后一个轴的输出维数，也就是颜色通道的数量相同。