423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2026-03-15 08:56:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

More type annotations

Browse Source 
Summary: More type annotations: device, shaders, pluggable I/O, stats in NeRF project, cameras, textures, etc...

Reviewed By: nikhilaravi

Differential Revision: D29327396

fbshipit-source-id: cdf0ceaaa010e22423088752688c8dd81f1acc3c
This commit is contained in:
Patrick Labatut
2021-06-25 19:55:26 -07:00
committed by Facebook GitHub Bot
parent 542e2e7c07
commit f593bfd3c2
15 changed files with 196 additions and 148 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pytorch3d/renderer/utils.py
View File

@@ -160,7 +160,7 @@ class TensorProperties(nn.Module):
msg = "Expected index of type int or slice; got %r"
raise ValueError(msg % type(index))
def to(self, device: Device = "cpu"):
def to(self, device: Device = "cpu") -> "TensorProperties":
"""
In place operation to move class properties which are tensors to a
specified device. If self has a property "device", update this as well.
@@ -174,7 +174,7 @@ class TensorProperties(nn.Module):
setattr(self, k, v.to(device_))
return self
def clone(self, other):
def clone(self, other) -> "TensorProperties":
"""
Update the tensor properties of other with the cloned properties of self.
"""
@@ -189,7 +189,7 @@ class TensorProperties(nn.Module):
setattr(other, k, v_clone)
return other
def gather_props(self, batch_idx):
def gather_props(self, batch_idx) -> "TensorProperties":
"""
This is an in place operation to reformat all tensor class attributes
based on a set of given indices using torch.gather. This is useful when