TensorFlow中的自定义丢失功能，带有额外参数

Question

我的损失函数不仅取决于标签和logits，还直接取决于其中一个输入变量。是否可以在TensorFlow中进行？还是其他任何框架？