found array with dim 3. estimator expected = 2. sklearn k clustering code example

Example: ValueError: Found array with dim 3. Estimator expected <= 2.

nsamples, nx, ny = train_dataset.shape
d2_train_dataset = train_dataset.reshape((nsamples,nx*ny))