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!