I can't understand if torch.scatter or torch.gather could be used to reduce values of a tensor acc