I have a tensor(A) of the Shape->  [1,3,13,13]
Another tensor(B) of the shape-> [3]
And a Threshold Value-> 0.5
I want to iterate over the -> X dimension of ->[1,X, 13, 13]
Eg. B= [0.1, 0.3, 0.6]
For each index of B[index], If its value > threshold make that [1,index, 13, 13] -> ZERO.
I tried with-> A(dim1, dim2> threshold, dim3, dim4) = 0
But it gives error-> [Index out of Bounds]
Any other way…