def METHOD_NAME(self) -> Dict[str, torch.Tensor]:
    results = self.compute()
    scalar_dict = {}
    for k, tensor_v in results.items():
        val = torch.flatten(tensor_v)
        if val.size(0) > 1:
            # non-scalar will be ignored
            continue
        scalar_dict[k] = val.item()
    return scalar_dict