Noah Marvy

segment_csr triggers shape error in backward pass when indptr does on cover src and reduction is mean

4

Minimum example ``` import torch_scatter import torch indptr = torch.tensor([0, 3, 7]) x = torch.randn(8, 11) lin = torch.nn.Linear(11, 11) opt = torch.optim.SGD(lin.parameters()) x = lin(x) x = torch_scatter.segment_csr(x, indptr,...