423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-21 14:50:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

lint fixes

Browse Source 
Summary:
Ran the linter.
TODO: need to update the linter as per D21353065.

Reviewed By: bottler

Differential Revision: D21362270

fbshipit-source-id: ad0e781de0a29f565ad25c43bc94a19b1828c020
This commit is contained in:
Nikhila Ravi
2020-05-04 09:55:03 -07:00
committed by Facebook GitHub Bot
parent 0c595dcf5b
commit 0eca74fa5f
15 changed files with 73 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pytorch3d/loss/chamfer.py
View File

@@ -117,8 +117,8 @@ def chamfer_distance(
P2 = y.shape[1]
# Check if inputs are heterogeneous and create a lengths mask.
is_x_heterogeneous = ~(x_lengths == P1).all()
is_y_heterogeneous = ~(y_lengths == P2).all()
is_x_heterogeneous = (x_lengths != P1).any()
is_y_heterogeneous = (y_lengths != P2).any()
x_mask = (
torch.arange(P1, device=x.device)[None] >= x_lengths[:, None]
) # shape [N, P1]