如何从任务列表(self.prediction)动态创建损失列表,而无需创建变量:
当前:
loss0 = tf.losses.softmax_cross_entropy( logits = self.prediction[0], onehot_labels = self.Y[0] ) # task 0
loss1 = tf.losses.softmax_cross_entropy( logits = self.prediction[1], onehot_labels = self.Y[1] ) # task 1
loss2 = tf.losses.softmax_cross_entropy( logits = self.prediction[2], onehot_labels = self.Y[2] ) # task 2
self.losses = tf.reduce_sum( [ loss0, loss1, loss2 ] )
目标:
list_loss = ?
self.losses = tf.reduce_sum( list_loss )
如果我明白你的意思,你是在要求
相关问题 更多 >
编程相关推荐