Skip to content

Commit bb8faaf

Browse files
committed
Remove useless clone
1 parent c1fc115 commit bb8faaf

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/torchjd/autogram/_engine.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,7 @@ def differentiation(_grad_output: Tensor) -> tuple[Tensor, ...]:
176176

177177
output_dims = list(range(output.ndim))
178178
identity = torch.eye(output.ndim, dtype=torch.int64)
179-
strides = torch.concatenate([identity, identity.clone()], dim=0)
179+
strides = torch.concatenate([identity, identity], dim=0)
180180
jac_output = make_sst(torch.ones_like(output), strides)
181181

182182
vmapped_diff = differentiation

0 commit comments

Comments
 (0)