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

fbcode/vision/fair/pytorch3d/pytorch3d/transforms/transform3d.py

Browse Source 
Reviewed By: bottler

Differential Revision: D93713606

fbshipit-source-id: a8aa52328a76d95d3985daec529cdce04ba12bd4
This commit is contained in:
generatedunixname1417043136753450
2026-02-22 07:06:34 -08:00
committed by meta-codesync[bot]
parent 1bd911d534
commit e70188ebbc
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pytorch3d/transforms/transform3d.py
View File

@@ -623,9 +623,7 @@ class Scale(Transform3d):
Return the inverse of self._matrix.
"""
xyz = torch.stack([self._matrix[:, i, i] for i in range(4)], dim=1)
# pyre-fixme[58]: `/` is not supported for operand types `float` and `Tensor`.
ixyz = 1.0 / xyz
# pyre-fixme[6]: For 1st param expected `Tensor` but got `float`.
ixyz = torch.reciprocal(xyz)
imat = torch.diag_embed(ixyz, dim1=1, dim2=2)
return imat