Thanks Roy,
3D would make sense in the case of batches of strings
for example…
(batch, sequence, character_embeddings)
I guess to do this in 2D, you would split the batch dimension and compute the loss one at a time?