s_ik is bascially one-hot vector which is 1 if data point i belongs to cluster k.
And for L2-reg. I simply want to implement Ridge Regression: Loss + \lambda || w ||_2.
where \lambda would be a hyperparameter and Loss = nn.mse().