xrg
commited on
Commit
•
5357224
1
Parent(s):
29c1deb
fix buf
Browse files- core/tensorBase.py +1 -1
core/tensorBase.py
CHANGED
@@ -286,7 +286,7 @@ class TensorBase(torch.nn.Module):
|
|
286 |
print(xyz_sampled.device)
|
287 |
print(self.aabb[0].device)
|
288 |
print(self.invaabbSize.device)
|
289 |
-
if xyz_sampled.device!=invaabbSize.device:
|
290 |
self.invaabbSize=self.invaabbSize.to(xyz_sampled.device)
|
291 |
|
292 |
return (xyz_sampled-self.aabb[0]) * self.invaabbSize - 1
|
|
|
286 |
print(xyz_sampled.device)
|
287 |
print(self.aabb[0].device)
|
288 |
print(self.invaabbSize.device)
|
289 |
+
if xyz_sampled.device!=self.invaabbSize.device:
|
290 |
self.invaabbSize=self.invaabbSize.to(xyz_sampled.device)
|
291 |
|
292 |
return (xyz_sampled-self.aabb[0]) * self.invaabbSize - 1
|