从遮罩创建张量

Question

处理一些自定义损失函数，我需要找到解决以下问题的方法。我对张量操纵不放心，因此非常感谢您的帮助：假设我有一些形状为A * B的布尔张量X。我想创建一个形状为A * B * 4的标量张量Y，使得：

Y[i,j,:] = [0,0,0,0] if X[i,j] = true
Y[i,j,:] = [c,c,0,0] if X[i,j] = false 

非常感谢！

（编辑：c是一些定义的常量）