def get_round_sum(tens, n_digits=3):
    val = torch.sum(tens, dim=int((not self.opt['attention_norm_idx'])))
    return ((val * (10 ** n_digits)).round() / (10 ** n_digits))
def get_round_sum(tens, n_digits=3):
    val = torch.sum(tens, dim=int((not self.opt['attention_norm_idx'])))
    return ((val * (10 ** n_digits)).round() / (10 ** n_digits))