That makes sense. I’m trying to create a custom LSTM cell, but I’m struggling to get template LSTM code I got from the forums here to work. Would you mind taking a quick look? Thanks! Size mismatch error when using custom LSTM cell 300