Using hybridblock I am writing hybrid_forward as follows:-
def hybrid_forward(self, F, x): if F.mean(x) > 1.0: x = x - 1 elif F.sum(x) < 0: x = F.abs(x) else: x = x + 1.0 return x
Above code shows error that “bool” is only implemented for NDArray, not for Symbol.
I know how to use mx.symbol.where, but I want to implement that in gluon and then hybridize it.
Any suggestions would be great.