init

pytorch · Nov 14, 2024 · 4d5b38c · 4d5b38c
1 parent db2b5e6
commit 4d5b38c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/tensordict/_td.py b/tensordict/_td.py
@@ -1026,6 +1026,10 @@ def reduction(val):
             )
 
         def reduction(val):
+            print(type(val), reduction_name)
+            print(val.device)
+            print(getattr(val, reduction_name))
+            print(kwargs)
             return getattr(val, reduction_name)(**kwargs)
 
         return self._fast_apply(

diff --git a/tensordict/base.py b/tensordict/base.py
@@ -10965,7 +10965,7 @@ def set_(x):
                     tgt = mb_unwrap_functional_tensor(new_thing)
                     src = mb_unwrap_functional_tensor(ragged_source)
                     tgt.nested_int_memo = src.nested_int_memo
-                else:
+                elif new_thing is not None:
                     _tensor_symint_registry[new_thing] = _tensor_symint_registry[
                         ragged_source
                     ]