Skip to content

Commit 8733909

Browse files
committed
[HOTFIX] Wrap torch abs
1 parent 02749cb commit 8733909

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

gempy_engine/core/backend_tensor.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -175,6 +175,7 @@ def _transpose(tensor, axes=None):
175175
cls.tfnp.concatenate = _concatenate
176176
cls.tfnp.transpose = _transpose
177177
cls.tfnp.geomspace = lambda start, stop, step: torch.logspace(start, stop, step, base=10)
178+
cls.tfnp.abs = lambda tensor, dtype: tensor.abs().type(dtype)
178179

179180
@classmethod
180181
def _wrap_pykeops_functions(cls):

0 commit comments

Comments
 (0)