casting numpy array into square array by comparing subarray

Question

I have and Array in shape (2000,256,2) and I want to cast it into the shape (2000,256,256) by comparing the sub arrays in the 3rd dimension.