implement Tensor._sis_hash

rwth-i6/returnn_common#252
rwth-i6 · Mar 10, 2023 · af1548a · af1548a
1 parent faf28c3
commit af1548a
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/returnn/tensor/_tensor_extra.py b/returnn/tensor/_tensor_extra.py
@@ -609,6 +609,15 @@ def __repr__(self):
     def __hash__(self):
         return id(self)
 
+    def _sis_hash(self):
+        if self.raw_tensor is not None and hasattr(self.raw_tensor, "_sis_hash"):
+            # noinspection PyProtectedMember
+            return self.raw_tensor._sis_hash()
+
+        from sisyphus.hash import sis_hash_helper  # noqa
+
+        return sis_hash_helper(self.get_kwargs())
+
     def __getstate__(self):
         d = {k: getattr(self, k) for k in self.__slots__}
         d["_raw_tensor"] = None  # do not store the TF tensors