Consider using tensordict as underlying structure for subjects

[fzimmermann89]

Hi,

as a long-term goal, it might be worthwhile to consider changing Subject to inherit from tensordict instead of dict. This would allow batching of subjects and might solve some of the issues observed in #1179

The downside is, that to make use tensorsdicts functionality, some work is needed and compatibility to existing code might suffer.

Cheers,
Felix

[fepegar]

Hi, @fzimmermann89. This sounds interesting.

This would allow batching of subjects

Can you please elaborate?

[fepegar]

FYI I tried implementing Subject and Image as TensorDicts, but that didn't solve the collating issue.

[romainVala]

too bad it looked promising