PyTorch Forums
Is this mask_index[draft_parents == 0] = -1 a scatter
Dave_Lee
(Dave Lee)
August 5, 2024, 6:12pm
1
mask_index[draft_parents == 0] = -1
Can this pytorch code be rewritten as torch.scatter operation?