Tensorflow Dataset To Numpy Array Slicing