'Device Side assert triggered at....' Error

[Kushaj (Kushajveer Singh)]

Also, what is the expected output you want by doing torch.cat(embeddings,1).