摘要: 问题1、自定义loss function,y_true shape多一个维度 def nce_loss(y_true, y_pred): y_true = tf.reshape(y_true, [-1]) y_true = tf.linalg.diag(y_true) ret = tf.keras. 阅读全文
posted @ 2021-11-12 10:04 hwYang 阅读(71) 评论(0) 推荐(0) 编辑