In my code, I want to replace values in the tensor given values of some indices are zero, for example
target_mac_out[av
