如何在Julia中为自定义函数定制广播（broadcast）方法？

Julia的广播机制默认会自动适配标量函数，但如果需要为自定义函数f实现特殊的广播逻辑，或者针对自定义类型定制行为，可以按以下方式操作：

1. 基础自动广播（无需额外代码） #

如果你的函数f已经实现了标量版本，Julia会自动支持f.(v)形式的广播调用，无需手动编写额外方法：

# 定义标量函数
f(x) = x^2 + 1

# 直接使用广播
v = [1, 2, 3]
f.(v)  # 输出 [2, 5, 10]

2. 自定义广播逻辑（针对特殊场景） #

如果需要为f的广播添加自定义行为（比如针对自定义类型、批量优化逻辑），核心是扩展Broadcast.broadcasted方法：

示例：针对自定义类型定制广播 #

假设你有一个自定义类型MyType，希望f.(MyType实例, 数组)采用批量处理逻辑而非逐个元素计算：

# 定义自定义类型
struct MyType
    val::Int
end

# 先实现标量版本的f
f(a::MyType, b::Int) = MyType(a.val + b)

# 导入broadcasted并扩展方法
import Base.Broadcast: broadcasted

function broadcasted(::typeof(f), a::MyType, b::AbstractArray{Int})
    # 批量处理逻辑：直接生成结果数组，替代默认的逐元素调用
    map(x -> MyType(a.val + x), b)
end

# 测试自定义广播
f.(MyType(1), [2, 3, 4])  # 输出 [MyType(3), MyType(4), MyType(5)]

3. 自定义广播返回的容器类型 #

如果需要让f.(v)返回特定的容器类型（比如自定义数组），可以扩展Broadcast.materialize方法，负责将广播中间对象转换为最终的容器：

# 定义自定义数组类型（需实现AbstractArray的必要接口）
struct MyArray{T} <: AbstractArray{T,1}
    data::Vector{T}
end

# 实现AbstractArray的基础接口示例
Base.size(a::MyArray) = size(a.data)
Base.getindex(a::MyArray, i::Int) = a.data[i]
Base.setindex!(a::MyArray, val, i::Int) = a.data[i] = val

# 导入materialize并扩展方法
import Base.Broadcast: materialize

function materialize(bc::Broadcast.Broadcasted{<:Any,<:Any,typeof(f),<:Tuple{MyArray}})
    # 将广播结果包装为MyArray类型
    MyArray(f.(bc.args[1].data))
end

# 测试自定义容器返回
arr = MyArray([1, 2, 3])
result = f.(arr)  # result是MyArray类型，包含[2,5,10]

内容的提问来源于stack exchange，提问作者Felix Benning