replace view with reshape, check for nans

Summary: Replace view with reshape, add check for nans before mesh sampling

Reviewed By: nikhilaravi

Differential Revision: D20548456

fbshipit-source-id: c4e1b88e033ecb8f0f3a8f3a33a04ce13a5b5043
This commit is contained in:
Georgia Gkioxari
2020-03-19 19:28:46 -07:00
committed by Facebook GitHub Bot
parent 53599770dd
commit 6c48ff6ad9
4 changed files with 25 additions and 3 deletions

View File

@@ -191,7 +191,7 @@ def padded_to_packed(
"Only one of split_size or pad_value should be provided."
)
x_packed = x.view(-1, D) # flatten padded
x_packed = x.reshape(-1, D) # flatten padded
if pad_value is None and split_size is None:
return x_packed