ThaiThien:

maybe this is a cause
thought_vectors[:, i, :] = output.clone()
try again with
thought_vectors[:, i, :] = output   # without .clone()
to see if it work.


No. I don’t think it is the reason.