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

Boolean indexing of cameras

Browse Source 
Summary: Reasonable to expect bool indexing.

Reviewed By: bottler, kjchalup

Differential Revision: D38741446

fbshipit-source-id: 22b607bf13110043c5624196c66ca1484fdbce6c
This commit is contained in:
Roman Shapovalov
2022-08-16 15:19:39 -07:00
committed by Facebook GitHub Bot
parent 60808972b8
commit b7c826b786
6 changed files with 58 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch3d/transforms/transform3d.py
View File

@@ -181,7 +181,7 @@ class Transform3d:
return self.get_matrix().shape[0]
def __getitem__(
self, index: Union[int, List[int], slice, torch.Tensor]
self, index: Union[int, List[int], slice, torch.BoolTensor, torch.LongTensor]
) -> "Transform3d":
"""
Args: