containers¶
NumberDict¶
- class NumberDict¶
just like collectionish.NumAttyDict but with a special format method for better handling in logging.
TensorDict¶
- class TensorDict¶
a very basic keyed attribute accessable container for tensors.
Can handle basic python math operators since it comes from
NumAttyDict
- to(device)¶
move all tensors in this Tensordict to device
- item()¶
get python numbers out of this Tensordict by calling item on all tensors in it.
Note
this will only work if all values in this tensordict are scalar tensors!
- Return type
NumAttyDict
- Returns