Any luck here? I have the same error.

Initially, restart the kernel.

Then check if you have done the initial part correct, i.e.

# Step 2.A: select the "t"th time step vector from X.

x = X[:,t,:]

Maybe this may solve your error.

@SwadhinNagulpelli That’s the answer for Exercise 1; @ladan_yang is asking about Exercise 2.

“Restart & *Clear Output*” under Kernel menu helps. Save work first just in case.

I was getting the same error message, while having solved the issue in exercise 1.

I solved it in exercise 2 by specifying the axis in the argmax function, setting it to 1. Then, all the expected shapes are consistent across the different steps. I also needed to restart the kernel and run all the cells to see the results of my solution.

x = tf.math.argmax(out, axis=1)

Hi, Sofisappia.

Welcome to the community.

Besides, it’s also worth to note that keeping the last axis in mind will also prove a significant step.