First, sharing your code is not allowed, so avoid sharing it.
Second, you posted this in the General Discussion category but, I guess, it belongs to DLS Course 1. Move it to the correct category as described here.
Third, you have to use dot
, not matmul
.
Fourth, you have to apply the correct activation function to Z
.