I guess you need to correct the second for-loop range in Positional Encoder. Because otherwise the statement pe[pos, i+1] would be accessing out of range memory for odd numbers