When I use torch.ge 6() function in network, I got following error:
‘Variable’ object is not callable
Code:
def forward(self, x):
     y = x.sum(dim = 1)
     yy = y.view(y.size(0), -1)
     avg = yy.mean(dim = 1)
     mask = torch.ge(y, avg(0))

where, x is an NCH*W tensor.