标签： python function tensorflow loss
我想在TensorFlow中为我的神经网络编写自定义损失函数。我很清楚如何编写简单的自定义损失函数，但是我希望损失函数取决于网络的预测值。我的网络输出0到1之间的值，并且如果输出大于0.95，我想应用另一个损失函数，而不是输出小于0.95的情况。我的问题是损失函数中的预测值只是张量而来，而我找不到解决方案来访问该张量的各个元素以查看必须应用哪个成本函数。感谢您的帮助！