`torch.unbind` doesn't seem work well with `defaultdict` since `defaultdict` doesn't cast the tensor into an `int` for indexing. Will `torch.unbind` followed by `.item()` during indexing be better, or will it be the same as `.tolist()`?