Fix StructuredSparseTensor.__new__

ValerianRey · ValerianRey · commit 7f2c9747d4e0 · 2025-11-06T15:53:41.000+01:00
diff --git a/src/torchjd/sparse/_structured_sparse_tensor.py b/src/torchjd/sparse/_structured_sparse_tensor.py
@@ -31,7 +31,7 @@ def __new__(cls, physical: Tensor, strides: Tensor):
         pshape = torch.tensor(physical.shape)
         vshape = strides @ (pshape - 1) + 1
         return Tensor._make_wrapper_subclass(
-            cls, vshape, dtype=physical.dtype, device=physical.device
+            cls, tuple(vshape.tolist()), dtype=physical.dtype, device=physical.device
         )
 
     def __init__(self, physical: Tensor, strides: Tensor):

Original file line number	Diff line number	Diff line change
`@@ -31,7 +31,7 @@ def __new__(cls, physical: Tensor, strides: Tensor):`
`31`	`31`	`pshape = torch.tensor(physical.shape)`
`32`	`32`	`vshape = strides @ (pshape - 1) + 1`
`33`	`33`	`return Tensor._make_wrapper_subclass(`
`34`		`- cls, vshape, dtype=physical.dtype, device=physical.device`
	`34`	`+ cls, tuple(vshape.tolist()), dtype=physical.dtype, device=physical.device`
`35`	`35`	`)`
`36`	`36`
`37`	`37`	`def __init__(self, physical: Tensor, strides: Tensor):`