diff --git a/src/Numerics.jl b/src/Numerics.jl index 2bd4a69..2128a4f 100644 --- a/src/Numerics.jl +++ b/src/Numerics.jl @@ -62,7 +62,7 @@ contract(a::Union{T,AbstractArray{T,0}}, b::Tensor{T}) where {T} = contract(Tens contract(a::Tensor{T}, b::Union{T,AbstractArray{T,0}}) where {T} = contract(a, Tensor(b)) contract(a::AbstractArray{<:Any,0}, b::AbstractArray{<:Any,0}) = contract(Tensor(a), Tensor(b)) |> only contract(a::Number, b::Number) = contract(fill(a), fill(b)) -contract(tensors::Tensor...) = reduce(contract, tensors) +contract(tensors::Tensor...; kwargs...) = reduce((x, y) -> contract(x, y; kwargs...), tensors) """ *(::Tensor, ::Tensor)