How to build TF tensor with ones in specified locations - batch compatible

Question

I apologize for the poor question title but I\'m not sure quite how to phrase it. Here\'s the problem I\'m trying to solve: I have two NNs working off of the same input data