C5_W1_A1 Ex 4: ValueError: could not broadcast input array from shape (5,10) into shape (2,10)

Hi everyone! I’m attempting the first assignment of Course 5 (Week 1) and I’m getting a ValueError in Exercise 4:

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-53-5b9326f2c416> in <module>
     14 parameters_tmp['by'] = np.random.randn(2, 1)
     15 
---> 16 a_tmp, y_tmp, c_tmp, caches_tmp = lstm_forward(x_tmp, a0_tmp, parameters_tmp)
     17 print("a[4][3][6] = ", a_tmp[4][3][6])
     18 print("a.shape = ", a_tmp.shape)

<ipython-input-52-c28b2f35d620> in lstm_forward(x, a0, parameters)
     55         a[:,:,t] = a_next
     56         # Save the value of the next cell state (≈1 line)
---> 57         c[:,:,t]  = c_next
     58         # Save the value of the prediction in y (≈1 line)
     59         y[:,:,t] = yt

ValueError: could not broadcast input array from shape (5,10) into shape (2,10)

Could someone please provide some guidance?

Nvm I figured it out. Initialized the tensor with an incorrect shape