caffe2中的条件性损失

Question

我想为caffe2中的两种数据集创建条件损失。每个数据集都有一个开关，因此任务是：

如果switch为True，则计算SigmoidCrossEntropyLoss，否则损失为0。

我对Caffe2并不是很熟悉，所以我不确定该怎么做。如果有人在Keras，Tensorflow或Pytorch等其他框架中做了类似的事情，我将很高兴听到您的建议并将其转换为caffe2。