arrays - 如何在 Julia 中乘以多维数组/矩阵

Question

将两个多维数组相乘，例如，一个 1 维数组与一个 3 维数组:

[1 2] * reshape(1:8,2,2,2)

给我错误信息:

LoadError: MethodError: `*` has no method matching *(::Array{Int64,2}, ::Array{Int64,3})
Closest candidates are:
  *(::Any, ::Any, !Matched::Any, !Matched::Any...)
  *{TA,TB}(::Union{DenseArray{TA,1},DenseArray{TA,2},SubArray{TA,1,A<:DenseArray{T,N},I<:Tuple{Vararg{Union{Colon,Int64,Range{Int64}}}},LD},SubArray{TA,2,A<:DenseArray{T,N},I<:Tuple{Vararg{Union{Colon,Int64,Range{Int64}}}},LD}}, !Matched::Base.LinAlg.AbstractTriangular{TB,S<:AbstractArray{T,2}})
  *{TA,TQ,N}(::Union{DenseArray{TA,N},SubArray{TA,N,A<:DenseArray{T,N},I<:Tuple{Vararg{Union{Colon,Int64,Range{Int64}}}},LD}}, !Matched::Union{Base.LinAlg.QRCompactWYQ{TQ,M<:AbstractArray{T,2}},Base.LinAlg.QRPackedQ{TQ,S<:AbstractArray{T,2}}})
  ...
while loading In[167], in expression starting on line 1

 in Ac_mul_B at operators.jl:157

将多维矩阵代数的数学定义用于矩阵/数组的 (1 x 2) * (2 x 2 x 2) 乘法。

一个更一般的例子可以是 A*B = C 表示 sum_k A_{i,j,k} B_{k,l,m} = C_{i,j,l,m}，其中 A 是 3 -索引矩阵，或者你喜欢的张量，B 是一个 3 索引矩阵，生成的 C 是一个四索引矩阵/张量，但是，一般来说，可以有任意数量的维度，并且维度可以有任意大小(在合理范围内)。查看有关 matrix product 定义的更多信息或 tensor contraction .

Julia 中这个乘法的正确语法是什么？

Answer 1

您可以使用reshape 将多维数组转换为矩阵， 将它们相乘，并将结果转换回多维数组。

A = [1 2]
B = reshape(1:8,2,2,2)
reshape( reshape(A,2,1)' * reshape(B,2,4), 2, 2 )

(在这个例子中，因为A已经是一个矩阵，所以实际上不需要reshape。)