One-hot mask a range of tensors given a label

Question

I am wondering how can I make use of torch.scatter_() or any other built-in masking function for this one-hot masking task-

I have two tensors-