Skip to content

Commit 93e3a61

Browse files
committed
Improve error message in cat_default
1 parent 7991ac1 commit 93e3a61

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

  • src/torchjd/sparse/_aten_function_overrides

src/torchjd/sparse/_aten_function_overrides/shape.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,7 @@ def cat_default(tensors: list[Tensor], dim: int) -> Tensor:
178178
if any(not torch.equal(t.strides, ref_strides) for t in tensors_[1:]):
179179
raise NotImplementedError(
180180
"Override for aten.cat.default does not support SSTs that do not all have the same "
181-
f"strides. Found the following strides:\n{[t.strides for t in tensors_]} and the "
181+
f"strides. Found the following tensors:\n{[t.debug_info() for t in tensors_]} and the "
182182
f"following dim: {dim}."
183183
)
184184

0 commit comments

Comments
 (0)