I need to compare two tensors element-wise and get a binary mask. e.g.
c = a > b
c = c.float()

However, this operation is not differentiable. Does someone know how I can calculate the element-wise> and still be able to backprop?