"Input dimension mis-match" in basic model?

You need a dot product here: