Flux/Lux - Combining two neural networks

Specific Domains Machine Learning
2

I think you want Parallel:

julia> m1 = Dense([1 2; 3 4]);

julia> m2 = Dense([5 6; 7 8]);

julia> x = [9, 10];

julia> m1(x)
2-element Vector{Int64}:
 29
 67

julia> m2(x)
2-element Vector{Int64}:
 105
 143

julia> m3 = Parallel(+, m1, m2)
Parallel(
  +,
  Dense(2 => 2),                        # 6 parameters
  Dense(2 => 2),                        # 6 parameters
)                   # Total: 4 arrays, 12 parameters, 352 bytes.

julia> m3(x)
2-element Vector{Int64}:
 134
 210
6 Likes