Hi,

I have an array “s_norm” with 120 entrys. For every entry a neural network should give me the values for these states

so as an output i have a matrix with 3 values for every s_norm (a 120x3 matrix)

I also have another array “a”, which is also 120 entrys long and has entrys between one and 3.

Now I want to produce an array, which is also 120 entrys long, but at every entry I want to have the values at position a for every s_norm.

I tried this:

but that gives me only the values of the firs column at the position a

Can you help me to get for every 120 columns the matching entry a?

probably something with squeeze to remove the dimensions, right?

That would be really nice! Thanks a lot!