The issue arose because you provided scalar values for the mean and variance, which assumed a single feature per sample.The error due to broadcasting, to address this, i correctly adjusted the code by passing lists of values, where each list contains the mean and variance for each of the five features per sample.Please refer to this gist to see changes i made