search

lambdaji / tf_repos

TensorFlow Script
681 stars 319 forks source link

code in the PNN.py file has little error #1

Closed liuzhiyong01 closed 6 years ago

liuzhiyong01 commented 6 years ago

outer = tf.reshape(tf.einsum('aik,ajk->aijk', p, q), [-1, num_pairsembedding_sizeembedding_size]) should be: outer = tf.reshape(tf.einsum('api,apj->apij', p, q), [-1, num_pairsembedding_sizeembedding_size])

lambdaji commented 6 years ago

谢谢纠正，outer PNN没经过测试，忘了注释掉