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

suppress errors in fbcode/vision - batch 2

Browse Source 
Differential Revision: D30222339

fbshipit-source-id: 97d498df72ef897b8dc2405764e3ffd432082e3c
This commit is contained in:
Pyre Bot Jr 2021-08-10 10:20:41 -07:00 committed by Facebook GitHub Bot
parent 804117833e
commit 9db70400d8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch3d/renderer/implicit/raysampling.py
View File

@ -123,7 +123,7 @@ class GridRaysampler(torch.nn.Module):
device = cameras.device
# expand the (H, W, 2) grid batch_size-times to (B, H, W, 2)
xy_grid = self._xy_grid.to(device)[None].expand( # pyre-ignore
xy_grid = self._xy_grid.to(device)[None].expand(
batch_size, *self._xy_grid.shape
)