Please help. I want to find the x,y coordinate of certain values in a tensor with size N, C, H, W
For example, I have a tensor as shown below:
N, C, H, W = 51, 1, 7, 7
x = torch.randn(N, C, H, W)
x_mean = x.view(N, -1).mean(1, keepdim=True) #calculating the mean for every batch
Now, I want to find the (x,y) locations in the 1, 7, 7
array where the values are greater than the x_mean
, for each batch.