Let’s say I have the tensor:
[22.2, 3.3, 3.2, 10.2, 3.2]
let’s also say that i have another tensor:
[0, 3, 1, 1, 3]
I’d like to turn the first tensor to:
[22.2, 0.0, 3.2, 10.2, 0.0]
That is where the second tensor’s value is 3, i want the first tensor’s value to be reset to 0.
If someone can write the 1-2 lines of c++ that corresponds to this it’d help me a lot.
Thank for the help.