Tensorflow prog. assign. Exercise 6

I got the following error code, could you please give a hint of what I did wrong?

Did you perhaps omit the transpose operations on the labels and logits. This was mentioned in the instructions for this section.

I added some prints to show the shapes and here’s what I see from that test cell:

logits.shape (2, 6)
labels.shape (2, 6)
tf.Tensor(0.810287, shape=(), dtype=float32)
All test passed

Here’s a recent thread that explains in more detail why the transpose is required.