julia - 为什么 julia 广播在处理多个数组时不起作用？

Question

我定义了两个结构体和一个这样的函数

struct A
    x::Float64
end

struct B
    y::Float64
end

f(a::A, b::B) = a.x*sin(b.y)
f.([A(0.1), A(0.2)], [B(1.), B(2.), B(3.)])

但是 f 返回这个错误:

DimensionMismatch("arrays could not be broadcast to a common size")

如何解决这个错误？我期望一个包含 6 个元素的数组作为函数输出。

Answer 1

问题是您的第一个参数是 2 元素 Vector，第二个参数是 3 元素 Vector。

如果你例如将第一个参数设为 1x2 Matrix，然后一切正常:

julia> f.([A(0.1) A(0.2)], [B(1.), B(2.), B(3.)])
3×2 Array{Float64,2}:
 0.0841471  0.168294
 0.0909297  0.181859
 0.014112   0.028224

(请注意，缺失的或长度为 1 的维度会自动广播)

请注意，您还可以广播对 A 和 B 构造函数的调用:

f.(A.([0.1 0.2]), B.(1.:3.))