fd55cf8ea1dd674fbe60ade84709fe5666dd9434,pointnet2_ops_lib/pointnet2_ops/pointnet2_utils.py,BallQuery,forward,#,245

Before Change

        torch.Tensor
            (B, npoint, nsample) tensor with the indicies of the features that form the query balls
        
        return _ext.ball_query(new_xyz, xyz, radius, nsample)

    @staticmethod
    def backward(ctx, a=None):
        return None, None, None, None

After Change

        torch.Tensor
            (B, npoint, nsample) tensor with the indicies of the features that form the query balls
        
        output = _ext.ball_query(new_xyz, xyz, radius, nsample)

        ctx.mark_non_differentiable(output)

        return output

    @staticmethod
    def backward(ctx, grad_out):
        return ()

Instances