File tree Expand file tree Collapse file tree 1 file changed +8
-4
lines changed
Expand file tree Collapse file tree 1 file changed +8
-4
lines changed Original file line number Diff line number Diff line change 2929⊗ () = Base. OneTo (1 )
3030
3131# Overload this version for most arrays
32- function fusedims (a:: AbstractArray , axes:: AbstractUnitRange... )
33- return fusedims (FusionStyle (a), a, axes... )
32+ function fusedims (a:: AbstractArray , ax :: AbstractUnitRange , axes:: AbstractUnitRange... )
33+ return fusedims (FusionStyle (a), a, ax, axes... )
3434end
3535
3636# Overload this version for fusion tensors, array maps, etc.
37- function fusedims (a:: AbstractArray , axesblocks:: Tuple{Vararg{AbstractUnitRange}} ...)
38- return fusedims (a, flatten_tuples (axesblocks)... )
37+ function fusedims (
38+ a:: AbstractArray ,
39+ axb:: Tuple{Vararg{AbstractUnitRange}} ,
40+ axesblocks:: Tuple{Vararg{AbstractUnitRange}} ...,
41+ )
42+ return fusedims (a, flatten_tuples ((axb, axesblocks... ))... )
3943end
4044
4145# Fix ambiguity issue
You can’t perform that action at this time.
0 commit comments