Skip to content
Snippets Groups Projects
Commit c6de524f authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Avoid depracation warning at::Tensor::data -> data_ptr

parent 664ee83b
Branches
Tags
No related merge requests found
...@@ -27,7 +27,7 @@ from pystencils_autodiff.tensorflow_jit import _hash ...@@ -27,7 +27,7 @@ from pystencils_autodiff.tensorflow_jit import _hash
class TorchTensorDestructuring(DestructuringBindingsForFieldClass): class TorchTensorDestructuring(DestructuringBindingsForFieldClass):
CLASS_TO_MEMBER_DICT = { CLASS_TO_MEMBER_DICT = {
FieldPointerSymbol: "data<{dtype}>()", FieldPointerSymbol: "data_ptr<{dtype}>()",
FieldShapeSymbol: "size({dim})", FieldShapeSymbol: "size({dim})",
FieldStrideSymbol: "strides()[{dim}]" FieldStrideSymbol: "strides()[{dim}]"
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment