I want to use Kmeans in the middle of my model. How can I do this? The output of this part of the model (ret) is as follows : size of ret is: 1 dimention of ret is: 0 shape of r
