Fix contract function

KeitaNakamura · Oct 16, 2024 · 09b5860 · 09b5860
1 parent e7511d0
commit 09b5860
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ops.jl b/src/ops.jl
@@ -146,7 +146,7 @@ end
     @assert length(xdims) == length(ydims)
     function create_indices(t, dims)
         indices = collect(1:ndims(t))
-        indices[dims] = 100 .+ (1:length(dims))
+        indices[dims] = (ndims(x)+ndims(y)) .+ (1:length(dims))
         Tuple(indices)
     end
     xindices = create_indices(x, xdims)