RuntimeError: Function _RasterizeGaussiansBackward returned an invalid gradient at index 5 - got [231550, 2] but expected shape compatible with [231550, 3]

[YuxuanWen-Code]

RuntimeError: Function _RasterizeGaussiansBackward returned an invalid gradient at index 5 - got [231550, 2] but expected shape compatible with [231550, 3]

I checked the input and the computed grad size, but it doesn't match:

Forward grad_scales shape: torch.Size([231550, 3]) [26/05 08:46:47]
Backward grad_scales shape: torch.Size([231550, 2]) [26/05 08:46:47]

What should I do to address this problem?

Many thanks!

[hbb1]

You should input tensor with 2 scales, as its name.