Hi Jonas,
sorry to hear you are trouble with named tensors.
To the best of my knowledge your approach should be correct and these methods are just not implemented yet.
Would the workaround work of returning a dict from your Dataset and add the names afterwards inside the DataLoader loop?