```bash return self.b + self.b_u[:,np.newaxis] + self.b_i[np.newaxis:,] + self.P.dot(self.Q.T) ``` ```bash self.b_i[np.newaxis:,] --> self.b_i[np.newaxis, :] ```
