implicit inplace operation '*=' cause an error when deriving the back gradient in pytorch

[Wei]

In the code Error, it cause the failure of deriving the back gradient.

Solution:
density_1_proxy = density_1_proxy*equals_one_mask[..., None]

[Phil Wang]

@VRCMF thanks Wei! 04201ee