Tensorflow Dataset To Numpy Array Reshape