我正在尝试自定义IoU丢失功能。我的模型有5个输出。第一个是0或1。如果框架中有一个对象，则为1，否则为0。其余4个输出为包围盒（回归）对于每个y_true，如果第一个输出为y_true的1，则它将计算IoU。否则，它将不会计算IoU。我无法遍历y_true / y_pred张量。我该如何实现？
def custom_loss(y_true, y_pred):
for y in y_true:
if first_index_value_is_iou:
calculate_IoU()