Yes, the second tensor contains values rather than int indexes. I basically want to make the first tensor have 0s in the same places the second tensor has 0s. I am not sure why the first tensor wouldnt be used. Here is an example:
rgb tensor:
tensor([
[
[ 23,  54],
[  1,  90]],
[
[ 22, 190],
[ 30,  10]
]])
Other tensor:
tensor([
[
[12,  0],
[ 0, 12]
]])
Resultant tensor:
tensor([
[
[ 23,  0],
[  0,  90]],
[
[ 22, 0],
[ 0,  10]
]])