It's hard to say to be honest mate. I really hope you have but wouldn't put money on it. I brought the iPad Air 2 through John Lewis in the UK. When i returned the first faulty unit the customer service adviser said that they had received word from Apple that any iPads with this fault should be exchanged.
Funny that when I returned the 2nd unit, the person I spoke to said he was not aware of such communication between themselves and Apple.
My biggest concern with this iPad is that it seems like its quite a relevant issue (put it this way, this thread alone is nearly at 40 pages).
Does not seem that there is a fix, but if I was to have a wild GUESS, I would say it's a hardware fault.