dimension does not match during SparseTensorDenseMatMul

[chouchou1988]

When running dssm/single/dssm_v3.py, InvalidArgumentError happens at SparseTensorDenseMatMul with the info "Cannot multiply A and B because inner dimension does not match". I check the code and find that maybe the shape for SparseTensorValue should be np.array([BS, TRIGRAM_D], np.int64) not np.array([BS, TRIGRAM_D], np.int64). Does anyone has the same problems?

[sehoi]

your question is not clear.
but try this.
change shape np.array([BS, TRIGRAM_D], np.int64) to np.array([None, TRIGRAM_D], np.int64).