Spaces:
Running
Running
Upload torchgeometry.diff
Browse files- patches/torchgeometry.diff +11 -0
patches/torchgeometry.diff
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
+++ venv/lib/python3.10/site-packages/torchgeometry/core/conversions.py
|
| 2 |
+
@@ -298,6 +298,9 @@
|
| 3 |
+
rmat_t[:, 0, 1] - rmat_t[:, 1, 0]], -1)
|
| 4 |
+
t3_rep = t3.repeat(4, 1).t()
|
| 5 |
+
|
| 6 |
+
+ mask_d2 = mask_d2.float()
|
| 7 |
+
+ mask_d0_d1 = mask_d0_d1.float()
|
| 8 |
+
+ mask_d0_nd1 = mask_d0_nd1.float()
|
| 9 |
+
mask_c0 = mask_d2 * mask_d0_d1
|
| 10 |
+
mask_c1 = mask_d2 * (1 - mask_d0_d1)
|
| 11 |
+
mask_c2 = (1 - mask_d2) * mask_d0_nd1
|