mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-01 03:12:49 +08:00
Moving ray bundle to float dtype
Summary: We can now move ray bundle to float dtype (e.g. from fp16 like types). Reviewed By: bottler Differential Revision: D57493109 fbshipit-source-id: 4e18a427e968b646fe5feafbff653811cd007981
This commit is contained in:
parent
bd52f4a408
commit
4ae25bfce7
@ -121,6 +121,19 @@ class ImplicitronRayBundle:
|
||||
else:
|
||||
self._lengths = value
|
||||
|
||||
def float_(self) -> None:
|
||||
"""Moves the tensors to float dtype in place
|
||||
(helpful for mixed-precision tensors).
|
||||
"""
|
||||
self.origins = self.origins.float()
|
||||
self.directions = self.directions.float()
|
||||
self._lengths = self._lengths.float() if self._lengths is not None else None
|
||||
self.xys = self.xys.float()
|
||||
self.bins = self.bins.float() if self.bins is not None else None
|
||||
self.pixel_radii_2d = (
|
||||
self.pixel_radii_2d.float() if self.pixel_radii_2d is not None else None
|
||||
)
|
||||
|
||||
def is_packed(self) -> bool:
|
||||
"""
|
||||
Returns whether the ImplicitronRayBundle carries data in packed state
|
||||
|
Loading…
x
Reference in New Issue
Block a user