Js Array Reduce Sum Tensorflow