Update new -> new_lietensor.

facebookresearch · Apr 27, 2023 · 3130d06 · 3130d06
1 parent 31e83b3
commit 3130d06
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/lie_labs.py b/examples/lie_labs.py
@@ -70,9 +70,9 @@ def is_shared(t1, t2):  # utility to check if memory is shared
 assert is_shared(g3_data, g6)
 
 # -- LieTensor.new()
-g7 = g3.new(g3_data)
-assert is_shared(g3_data, g7)  # shares storage
-assert g7.grad_fn is not None  # differentiable
+g7 = g3.new_lietensor(g3_data)
+assert not is_shared(g3_data, g7)  # doesn't share storage
+assert g7.grad_fn is None  # creates a leaf
 
 # ### Lie operations
 v = torch.randn(batch_size, 6)