Set specific channel to zero

2

you are multiplying by the value 1. Because mask is torch.ones.
The value will not change in that case, as expected.