In your code ```python raw_pt = pt[:,:3,:].contiguous() normal = pt[:,3:,:].transpose(1, 2).contiguous() if C > 3 else None ``` Therefore, I was wondering do you used the normal feature except xyz coordinate to further boost the performance?