Yet another shape issue in C3_W1_E6

[arvyzukai]

Hi @Karan_Abrol

Have you checked this post?

Cheers

P.S. Concrete example regarding the shapes (here batch=1):

inputs:
DeviceArray([[   2, 2378, 2939]], dtype=int32)

# Embedding_9088_256 - outputs.shape:
(1, 3, 256)

# Mean - outputs.shape:
(1, 256)

# Dense_2 - outputs.shape
(1, 2)

outputs:
DeviceArray([[ 0.548059 , -0.5545566]], dtype=float32)