Skip to content

Commit 3591e19

Browse files
committed
STY: no explicit result=... variable
1 parent 862c12d commit 3591e19

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

torch_np/_funcs.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1227,9 +1227,7 @@ def vdot(a: ArrayLike, b: ArrayLike, /):
12271227
def tensordot(a: ArrayLike, b: ArrayLike, axes=2):
12281228
if isinstance(axes, (list, tuple)):
12291229
axes = [[ax] if isinstance(ax, int) else ax for ax in axes]
1230-
result = torch.tensordot(a, b, dims=axes)
1231-
1232-
return result
1230+
return torch.tensordot(a, b, dims=axes)
12331231

12341232

12351233
@normalizer

0 commit comments

Comments
 (0)