FoldGather axis=0: shape [1,3] values = [3, 0, 0]   reference numpy.take(params,[1],axis=0) = [3, 4, 5]
MISMATCH
