I’m not sure where does it wrong.
I solved it! I used np.dot instead of * .
Well done @Mayu! As you realized ‘*’ is element-wise multiplication, which is not the right kind of multiplication in this case. We need to use matrix multiplication either by using functions or the ‘@’ (at) operator.
I see! Thanks, @CatalinaAlbornoz!