i have as a input for my neural network basically an array of arrays. Each of these arrays should have seperate input neurons associated to them. A small example can be found below, where my training set has size two and i want to have 3 input heads. This code yields an error since the input for the training has the wrong shape. How do i need to process the input correctly?
using Flux m= Chain(Parallel(+; head1 = Dense(4, 2, tanh), head2 = Dense(2, 2), head3 = Dense(3, 2, tanh)),Dense(2,1,tanh)) inputs = [([1,2,3],[4,5], [50,100, 80]),([10,20,30],[40,50], [70,130, 80])] outputs = randn((1,2)) size(outputs) data = [(inputs, outputs)] loss(x, y) = Flux.Losses.mse(m(x), y) ps = Flux.params(m) # later opt = Descent(0.1) Flux.train!(loss, ps, data, opt)