@alexis-jacq Thank you for your response!
All is ok with shapes, it works. The question about gradient, does it flow correctly? It seems that not because the model converges with batch size = 1 but withbatch size > 1 it doesn’t. Can someone help?
Also, my decoder is Many-to-One LSTM so it is not standard seq2seq and my labels have shape = [batch_size] so I don’t need to mask labels
Thanks