What do you mean by “save”?
If you would like to mask the images at the False positions, you could simply multiply both tensors:
mask = torch.randn(2, 4, 4) > 0.
imgs = torch.randn(2, 4, 4)

masked_imgs = imgs * mask