ValueError: Dimension must be 4 but is 3 for 'capsule_2/transpose_4' (op: 'Transpose') with input shapes: [?,10,16,16], [3]

Question

I am trying to use capsule networks for binary text classification. I use pre-trained fasttext word embedding to vectorize my dataset

, and this is my model definition